Update app.py

app.py

@@ -18,6 +18,7 @@ def excute_udiff(diffusion_model_id, concept, attacker):
     print(f"my IP is {myip}, my port is {myport}")
     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})
+    result = json.load(result.text)["prompt"]
     
     return f"The attack prompt is: {result}"
 
@@ -51,7 +52,7 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Column():
         gr.Examples(examples=[
-            ["runwayml/stable-diffusion-v1-5", "nudity", "text_grad"]
+            ["CompVis/stable-diffusion-v1-4", "nudity", "text_grad"]
         ], inputs=[diffusion_model_id, concept, attacker])
 
     start_button.click(fn=excute_udiff, inputs=[diffusion_model_id, concept, attacker], outputs=result, api_name="udiff")