Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -222,7 +222,7 @@ def process_images(
|
|
| 222 |
@spaces.GPU
|
| 223 |
def get_text_embeddings(concept_file):
|
| 224 |
descriptions = load_descriptions(concept_file)
|
| 225 |
-
embeddings = generate_embeddings(descriptions,
|
| 226 |
return embeddings, True
|
| 227 |
|
| 228 |
|
|
|
|
| 222 |
@spaces.GPU
|
| 223 |
def get_text_embeddings(concept_file):
|
| 224 |
descriptions = load_descriptions(concept_file)
|
| 225 |
+
embeddings = generate_embeddings(descriptions, clip_model, tokenizer, device, batch_size=100)
|
| 226 |
return embeddings, True
|
| 227 |
|
| 228 |
|