image-embeddings

Running

AkinyemiAra commited on 13 days ago

Commit

db457c9

verified ·

1 Parent(s): 28ee9f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ def ImgEmbed(image: Image.Image):
     img_emb = vision_model(**inputs).last_hidden_state
     img_embeddings = F.normalize(img_emb[:, 0], p=2, dim=1)
-    return img_embeddings[0].tolist(), focused_image
 # Gradio UI
 with gr.Blocks() as demo:

     img_emb = vision_model(**inputs).last_hidden_state
     img_embeddings = F.normalize(img_emb[:, 0], p=2, dim=1)
+    return img_embeddings[0].tolist()
 # Gradio UI
 with gr.Blocks() as demo: