Update app.py

app.py

@@ -38,7 +38,7 @@ def find_similarity(base64_image, text_input):
     text_features /= text_features.norm(dim=-1, keepdim=True)
     similarity = (text_features @ image_features.T).squeeze(0).cpu().numpy()
 
-    return similarity[0, 0]
+    return similarity
 
 # Create a Gradio interface
 iface = gr.Interface(