TuringsSolutions
commited on
Commit
•
6c8bc81
1
Parent(s):
c537534
Update app.py
Browse files
app.py
CHANGED
@@ -172,11 +172,11 @@ def generate_new_image():
|
|
172 |
interface = gr.Interface(
|
173 |
fn=train_snn,
|
174 |
inputs=[
|
175 |
-
gr.
|
176 |
-
gr.
|
177 |
-
gr.
|
178 |
],
|
179 |
-
outputs=gr.
|
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 |
)
|