Update app.py

app.py

@@ -172,13 +172,13 @@ def generate_new_image():
 interface = gr.Interface(
     fn=train_snn,
     inputs=[
-        gr.components.Image(type="pil", label="Upload Target Image"),
-        gr.components.Slider(minimum=500, maximum=3000, default=2000, label="Number of Agents"),
-        gr.components.Slider(minimum=10, maximum=200, default=100, label="Number of Epochs")
+        gr.Image(type="pil", label="Upload Target Image"),
+        gr.Slider(minimum=500, maximum=3000, value=2000, label="Number of Agents"),
+        gr.Slider(minimum=10, maximum=200, value=100, label="Number of Epochs")
     ],
-    outputs=gr.components.Image(type="numpy", label="Generated Image"),
+    outputs=gr.Image(type="numpy", label="Generated Image"),
     title="Swarm Neural Network Image Generation",
     description="Upload an image and set the number of agents and epochs to train the Swarm Neural Network to generate a new image."
 )
 
-interface.launch()
+interface.launch()