Update app.py

app.py

@@ -97,17 +97,13 @@ def get_recommendations(img, n_neighbors=5):
     recommended_model_names = [model_names[idx] for idx in indices.flatten()]
     recommended_distances = distances.flatten()
 
-    return [Image.open(img_path) for img_path in recommended_images], recommended_model_names, recommended_distances
+    return [(Image.open(img_path), f'{name}, Distance: {dist:.2f}') for img_path, name, dist in zip(recommended_images, recommended_model_names, recommended_distances)]
 
 # Gradio interface
-def display_images(input_image):
-    recommended_images, recommended_model_names, recommended_distances = get_recommendations(input_image)
-    return [(img, f'{name}, Distance: {dist:.2f}') for img, name, dist in zip(recommended_images, recommended_model_names, recommended_distances)]
-
 interface = gr.Interface(
-    fn=display_images,
+    fn=get_recommendations,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Carousel(label="Recommended Images", item_shape=(256, 256)),
+    outputs=gr.Gallery(label="Recommended Images"),
     title="Image Recommendation System",
     description="Upload an image and get similar images with their model names and distances."
 )