Update app.py

app.py

@@ -34,10 +34,10 @@ def process_image_from_binary(img_stream):
     
     return img
 
-def generate_img(concept, prompt):
+def generate_img(concept, prompt, seed, steps):
     print(f"my IP is {myip}, my port is {myport}")
     response = requests.post('http://{}:{}/generate'.format(myip, myport), 
-                             json={"concept": concept, "prompt": prompt},
+                             json={"concept": concept, "prompt": prompt, "seed": seed, "steps": steps},
                              timeout=(10, 1200))
     print(f"result: {response}")
     image = None
@@ -73,6 +73,6 @@ with gr.Blocks() as demo:
         with gr.Column(min_width=512):
             result_img = gr.Image(label="Image Gnerated by AdvUnlearn",width=512,show_share_button=False,show_download_button=False)
 
-    start_button.click(fn=generate_img, inputs=[drop_text, text_input], outputs=result_img, api_name="generate")
+    start_button.click(fn=generate_img, inputs=[drop_text, text_input, seed, steps], outputs=result_img, api_name="generate")
             
 demo.launch()