spidey121 commited on
Commit
b9cc85e
·
1 Parent(s): 7cdbef9

fix startup order

Browse files
Files changed (3) hide show
  1. Dockerfile +1 -1
  2. env/fake_server.py +0 -9
  3. inference.py +3 -0
Dockerfile CHANGED
@@ -8,4 +8,4 @@ RUN pip install --no-cache-dir -r requirements.txt
8
 
9
  EXPOSE 7860
10
 
11
- CMD ["python", "app.py"]
 
8
 
9
  EXPOSE 7860
10
 
11
+ CMD ["python", "inference.py"]
env/fake_server.py CHANGED
@@ -171,16 +171,7 @@ def home():
171
 
172
  # ---------------- Run Server ----------------
173
 
174
- def start_inference():
175
- import time
176
- time.sleep(5) # wait for server start
177
- import inference
178
-
179
  def run_server():
180
- import threading
181
-
182
- threading.Thread(target=start_inference, daemon=True).start()
183
-
184
  app.run(
185
  host="0.0.0.0",
186
  port=7860,
 
171
 
172
  # ---------------- Run Server ----------------
173
 
 
 
 
 
 
174
  def run_server():
 
 
 
 
175
  app.run(
176
  host="0.0.0.0",
177
  port=7860,
inference.py CHANGED
@@ -7,6 +7,9 @@ from openai import OpenAI
7
  random.seed(42)
8
 
9
  from env.fake_server import run_server
 
 
 
10
  from env.env import DeceptionEnv
11
  from env.attacker import simulate_attack
12
 
 
7
  random.seed(42)
8
 
9
  from env.fake_server import run_server
10
+ threading.Thread(target=run_server, daemon=True).start()
11
+
12
+ time.sleep(3)
13
  from env.env import DeceptionEnv
14
  from env.attacker import simulate_attack
15