linoyts HF Staff commited on
Commit
747633e
·
verified ·
1 Parent(s): 4375f4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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("done with text embeddings")
227
- return embeddings, gr.update(value=True)
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(