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 = 100
+subset_size = 50
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images
@@ -145,9 +145,9 @@ def display_images(image_paths, model_names, distances):
 interface = gr.Interface(
     fn=recommend,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Textbox(),  # Use Textbox for better text display
+    outputs=gr.Textbox(),
     title="Image Recommendation System",
     description="Upload an image and get 5 recommended similar images with model names and distances."
 )
 
-interface.launch(share=True)
+interface.launch()