Spaces:
Runtime error
Runtime error
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.
|