Jiqing commited on
Commit
a97a3a1
1 Parent(s): c7a8c4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def excute_udiff(diffusion_model_id, concept, attacker):
18
  print(f"my IP is {myip}, my port is {myport}")
19
  print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, attacker: {attacker}")
20
  result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "attacker": attacker})
21
- result = json.load(result.text)["prompt"]
22
 
23
  return f"The attack prompt is: {result}"
24
 
 
18
  print(f"my IP is {myip}, my port is {myport}")
19
  print(f"my input is diffusion_model_id: {diffusion_model_id}, concept: {concept}, attacker: {attacker}")
20
  result = requests.post('http://{}:{}/udiff'.format(myip, myport), json={"diffusion_model_id": diffusion_model_id, "concept": concept, "attacker": attacker})
21
+ result = result.text["prompt"]
22
 
23
  return f"The attack prompt is: {result}"
24