Commit
·
95f47a6
1
Parent(s):
3525ad6
Add running event loop
Browse files- src/streamlit_app.py +8 -0
src/streamlit_app.py
CHANGED
@@ -1,4 +1,6 @@
|
|
1 |
import os
|
|
|
|
|
2 |
import tempfile
|
3 |
|
4 |
os.environ["HOME"] = "/tmp"
|
@@ -7,6 +9,12 @@ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
|
7 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
8 |
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
|
9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
import cv2
|
11 |
import numpy as np
|
12 |
from PIL import Image
|
|
|
1 |
import os
|
2 |
+
import sys
|
3 |
+
import asyncio
|
4 |
import tempfile
|
5 |
|
6 |
os.environ["HOME"] = "/tmp"
|
|
|
9 |
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
10 |
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
|
11 |
|
12 |
+
if sys.platform.startswith('linux'):
|
13 |
+
try:
|
14 |
+
asyncio.get_event_loop()
|
15 |
+
except RuntimeError:
|
16 |
+
asyncio.set_event_loop(asyncio.new_event_loop())
|
17 |
+
|
18 |
import cv2
|
19 |
import numpy as np
|
20 |
from PIL import Image
|