spidey121 commited on
Commit
96e12d5
·
1 Parent(s): da6422e

fix round 1

Browse files
Files changed (1) hide show
  1. inference.py +8 -2
inference.py CHANGED
@@ -8,8 +8,14 @@ from openai import OpenAI
8
  random.seed(42)
9
 
10
  from env.fake_server import run_server
11
- threading.Thread(target=run_server, daemon=True).start()
12
- time.sleep(2)
 
 
 
 
 
 
13
  from env.env import DeceptionEnv
14
  from env.attacker import simulate_attack
15
 
 
8
  random.seed(42)
9
 
10
  from env.fake_server import run_server
11
+ def start_server():
12
+ try:
13
+ requests.get("http://127.0.0.1:7860/status", timeout=1)
14
+ # server already running
15
+ except:
16
+ threading.Thread(target=run_server, daemon=True).start()
17
+ time.sleep(2)
18
+ start_server()
19
  from env.env import DeceptionEnv
20
  from env.attacker import simulate_attack
21