ktllc commited on
Commit
15470aa
·
1 Parent(s): 82eb2a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def find_similarity(base64_image, text_input):
33
  # Calculate cosine similarity
34
  similarity = (image_features @ text_features.T).squeeze(0).cpu().numpy()
35
 
36
- return similarity
37
 
38
  # Create a Gradio interface
39
  iface = gr.Interface(
 
33
  # Calculate cosine similarity
34
  similarity = (image_features @ text_features.T).squeeze(0).cpu().numpy()
35
 
36
+ return similarity[0,0]
37
 
38
  # Create a Gradio interface
39
  iface = gr.Interface(