linoyts HF Staff commited on
Commit
ee8f1f1
·
verified ·
1 Parent(s): 07c2dd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, model, tokenizer, device, batch_size=100)
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