Update app.py

app.py

@@ -10,11 +10,7 @@ import cv2
 import itertools
 import sys
 
-os.system("sys.path.insert(0, 'home/user/app/test_detection/yolov5')")
-sys.path.insert(0, 'home/user/app/test_detection/yolov5')
-  
-os.system("sys.path.insert(0, 'home/user/app/test_detection/yolov5/models')")
-sys.path.insert(0, 'home/user/app/test_detection/yolov5/models')
+
   
 def load_model():
     wpath = 'test_detection/yolov5/weights/crowdhuman_yolov5m.pt'
@@ -127,6 +123,15 @@ if __name__ == '__main__':
   sys.path.insert(0, './app/test_detection/yolov5')
   sys.path.insert(0, '/app/test_detection/yolov5')
   
+  os.environ['PYTHONPATH'] += "home/user/app/test_detection/yolov5"
+  os.environ['PYTHONPATH'] += "test_detection/yolov5/models"
+
+  os.system("sys.path.insert(0, 'home/user/app/test_detection/yolov5')")
+  sys.path.insert(0, 'home/user/app/test_detection/yolov5')
+  
+  os.system("sys.path.insert(0, 'home/user/app/test_detection/yolov5/models')")
+  sys.path.insert(0, 'home/user/app/test_detection/yolov5/models')
+
 
 
   st.write(os.listdir('test_detection/yolov5/'))