spidey121 commited on
Commit
c883ba0
·
1 Parent(s): 424dfc5

fix round 1

Browse files
Files changed (1) hide show
  1. inference.py +10 -2
inference.py CHANGED
@@ -8,8 +8,16 @@ 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/")
14
+ print("Server already running")
15
+ except:
16
+ threading.Thread(target=run_server, daemon=True).start()
17
+ time.sleep(2)
18
+
19
+
20
+ start_server()
21
  from env.env import DeceptionEnv
22
  from env.attacker import simulate_attack
23