Update app.py

app.py

@@ -16,7 +16,7 @@ import matplotlib.pyplot as plt
 dataset = load_dataset("thefcraft/civitai-stable-diffusion-337k")
 
 # Take a subset of the dataset
-subset_size = 4000
+subset_size = 100
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images
@@ -137,13 +137,14 @@ def display_images(image_paths, model_names, distances):
         plt.axis('off')
     plt.show()
 
-# Gradio interface
 interface = gr.Interface(
     fn=recommend,
-    inputs=gr.inputs.Image(type="pil"),
-    outputs="text",  # Outputs the list of recommended images, models, and distances
+    inputs=gr.Image(type="pil"),  # Updated input component
+    outputs=gr.Text(),  # Updated output component
     title="Image Recommendation System",
     description="Upload an image and get 5 recommended similar images with model names and distances."
 )
 
+interface.launch()
+
 interface.launch()