Update app.py

app.py

@@ -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
+    return img_embeddings[0].tolist()
 
 # Gradio UI
 with gr.Blocks() as demo: