Update app.py

app.py

@@ -108,17 +108,21 @@ def display_images(image_paths, model_names, distances):
     plt.show()
 
 def recommend_images(img):
-  recommended_images, recommended_model_names, recommended_distances = get_recommendations(img, model, nbrs, image_paths, model_names)
-  recommendations = []
-  for i in range(len(recommended_images)):
-    recommendations.append(f"Rekomendasi {i+1}:\nGambar: {Image.open(recommended_images[i])}\nModel: {recommended_model_names[i]}\nEuclidian Distance: {recommended_distances[i]:.2f}")
-  return recommendations
+    temp_img_path = 'temp_img.jpg'
+    img.save(temp_img_path)
+    
+    recommended_images, recommended_model_names, recommended_distances = get_recommendations(temp_img_path, model, nbrs, image_paths, model_names)
+    
+    recommendations = []
+    for i in range(len(recommended_images)):
+        recommendations.append(f"Rekomendasi {i+1}:\nGambar: {recommended_images[i]}\nModel: {recommended_model_names[i]}\nEuclidian Distance: {recommended_distances[i]:.2f}")
+    return recommended_images
 
 iface = gr.Interface(
     fn=recommend_images,
     inputs=gr.Image(label="Upload an image"),
-    outputs=gr.Textbox(label="Rekomendasi", lines=10),
+    outputs=gr.Gallery(label="Rekomendasi"),
     title="Image Recommendation System",
     description="Upload an image and get recommendations based on similarity."
 )
-iface.launch()
+iface.launch()