Spaces:
Sleeping
Sleeping
add bits for ZeroGPU operation
Browse files
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.
|