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 = 50
+subset_size = 1000
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images
@@ -136,4 +135,5 @@ iface = gr.Interface(
 )
 
 # Launch the Gradio app
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()