omar87 commited on
Commit
fd96822
·
1 Parent(s): dc979d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,8 @@ def image_similarity(target_image, image_list):
30
  scores = compute_similarity(target_image, image_list)
31
  results = list(zip(image_list, scores))
32
  results.sort(key=lambda x: x[1], reverse=True) # Sort by similarity score
33
- return [str(result) for result in results]
 
34
 
35
  # Prepare Gradio interface
36
  iface = gr.Interface(
 
30
  scores = compute_similarity(target_image, image_list)
31
  results = list(zip(image_list, scores))
32
  results.sort(key=lambda x: x[1], reverse=True) # Sort by similarity score
33
+ formatted_results = [(str(image.tolist()), f"{score:.4f}") for image, score in results]
34
+ return formatted_results
35
 
36
  # Prepare Gradio interface
37
  iface = gr.Interface(