TuringsSolutions commited on
Commit
6c8bc81
1 Parent(s): c537534

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -172,11 +172,11 @@ def generate_new_image():
172
  interface = gr.Interface(
173
  fn=train_snn,
174
  inputs=[
175
- gr.inputs.Image(type="pil", label="Upload Target Image"),
176
- gr.inputs.Slider(minimum=500, maximum=3000, default=2000, label="Number of Agents"),
177
- gr.inputs.Slider(minimum=10, maximum=200, default=100, label="Number of Epochs")
178
  ],
179
- outputs=gr.outputs.Image(type="numpy", label="Generated Image"),
180
  title="Swarm Neural Network Image Generation",
181
  description="Upload an image and set the number of agents and epochs to train the Swarm Neural Network to generate a new image."
182
  )
 
172
  interface = gr.Interface(
173
  fn=train_snn,
174
  inputs=[
175
+ gr.components.Image(type="pil", label="Upload Target Image"),
176
+ gr.components.Slider(minimum=500, maximum=3000, default=2000, label="Number of Agents"),
177
+ gr.components.Slider(minimum=10, maximum=200, default=100, label="Number of Epochs")
178
  ],
179
+ outputs=gr.components.Image(type="numpy", label="Generated Image"),
180
  title="Swarm Neural Network Image Generation",
181
  description="Upload an image and set the number of agents and epochs to train the Swarm Neural Network to generate a new image."
182
  )