Spaces:
Runtime error
Runtime error
fix startup order
Browse files- Dockerfile +1 -1
- env/fake_server.py +0 -9
- 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", "
|
|
|
|
| 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 |
|