RoboApocalypse commited on
Commit
3352f0c
·
verified ·
1 Parent(s): 9ec26d1

add bits for ZeroGPU operation

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,6 +20,7 @@ model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
20
 
21
 
22
  # Define function to generate text embeddings
 
23
  def generate_text_embedding(text_data: Union[str, tuple[str]]) -> list[str]:
24
  """
25
  Generate embeddings for text data using the OpenCLIP model.
@@ -75,6 +76,7 @@ def generate_text_embedding(text_data: Union[str, tuple[str]]) -> list[str]:
75
  return text_embeddings
76
 
77
  # Define function to generate image embeddings
 
78
  def generate_image_embedding(image_data: Union[Image.Image, tuple[Image.Image]]) -> list[str]:
79
  """
80
  Generate embeddings for image data using the OpenCLIP model.
 
20
 
21
 
22
  # Define function to generate text embeddings
23
+ @spaces.GPU
24
  def generate_text_embedding(text_data: Union[str, tuple[str]]) -> list[str]:
25
  """
26
  Generate embeddings for text data using the OpenCLIP model.
 
76
  return text_embeddings
77
 
78
  # Define function to generate image embeddings
79
+ @spaces.GPU
80
  def generate_image_embedding(image_data: Union[Image.Image, tuple[Image.Image]]) -> list[str]:
81
  """
82
  Generate embeddings for image data using the OpenCLIP model.