Update app.py

app.py

@@ -18,9 +18,9 @@ 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 = result.text["prompt"]
+    result = result.text[1:-1]
     
-    return f"The attack prompt is: {result}"
+    return f"The unsafe prompt is: {result}"
 
 
 css = '''