Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -221,10 +221,11 @@ def process_images(
|
|
221 |
|
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 |
-
print("
|
227 |
-
return embeddings,
|
228 |
|
229 |
|
230 |
def process_and_display(
|
|
|
221 |
|
222 |
@spaces.GPU
|
223 |
def get_text_embeddings(concept_file):
|
224 |
+
print("generating text embeddings")
|
225 |
descriptions = load_descriptions(concept_file)
|
226 |
embeddings = generate_embeddings(descriptions, clip_model, tokenizer, device, batch_size=100)
|
227 |
+
print("text embeddings shape",embeddings.shape)
|
228 |
+
return embeddings, True
|
229 |
|
230 |
|
231 |
def process_and_display(
|