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 |
|