Update app.py

app.py

@@ -14,10 +14,10 @@ is_spaces = True if "SPACE_ID" in os.environ else False
 is_shared_ui = False
 
 
-def excute_udiff(model_id, concept, attacker):
+def excute_udiff(diffusion_model_id, concept, attacker):
     print(f"my IP is {myip}, my port is {myport}")
-    print(f"my input is model_id: {model_id}, concept: {concept}, attacker: {attacker}")
-    result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": model_id, "concept": concept, "attacker": attacker})
+    print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, attacker: {attacker}")
+    result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "attacker": attacker})
     
     return f"The attack prompt is: {result}"
 
@@ -40,7 +40,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row() as udiff:
         with gr.Column():
             # gr.Markdown("Please upload your model id.")
-            model_id = gr.Textbox(label='model_id')
+            diffusion_model_id = gr.Textbox(label='diffusion_model_id')
             concept = gr.Textbox(label='concept')
             attacker = gr.Textbox(label='attacker')
 
@@ -52,9 +52,9 @@ with gr.Blocks(css=css) as demo:
     with gr.Column():
         gr.Examples(examples=[
             ["runwayml/stable-diffusion-v1-5", "nudity", "text_grad"]
-        ], inputs=[model_id, concept, attacker])
+        ], inputs=[diffusion_model_id, concept, attacker])
 
-    start_button.click(fn=excute_udiff, inputs=[model_id, concept, attacker], outputs=result, api_name="udiff")
+    start_button.click(fn=excute_udiff, inputs=[diffusion_model_id, concept, attacker], outputs=result, api_name="udiff")
 
 
 # demo.queue(default_enabled=False, api_open=False, max_size=5).launch(debug=True, show_api=False)