sudo-paras-shah commited on
Commit
95f47a6
·
1 Parent(s): 3525ad6

Add running event loop

Browse files
Files changed (1) hide show
  1. 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