Update app.py

app.py

@@ -8,14 +8,13 @@ from tensorflow.keras.preprocessing import image
 from sklearn.neighbors import NearestNeighbors
 import joblib
 from PIL import UnidentifiedImageError, Image
-import matplotlib.pyplot as plt
 import gradio as gr
 
 # Load the dataset
 dataset = load_dataset("thefcraft/civitai-stable-diffusion-337k")
 
 # Take a subset of the dataset
-subset_size = 100  # Reduce the subset size for faster execution in Spaces
+subset_size = 1000  # Reduce the subset size for faster execution in Spaces
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images
@@ -107,8 +106,8 @@ def display_images(input_image):
 
 interface = gr.Interface(
     fn=display_images,
-    inputs=gr.inputs.Image(type="pil"),
-    outputs=gr.outputs.Carousel(label="Recommended Images", item_shape=(256, 256)),
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Carousel(label="Recommended Images", item_shape=(256, 256)),
     title="Image Recommendation System",
     description="Upload an image and get similar images with their model names and distances."
 )