Update app.py

app.py

@@ -80,11 +80,12 @@ def starter():
     
 def prediction(vname):
     
+    
     vpath='test_detection/data/'+vname
     wpath = 'test_detection/yolov5/weights/crowdhuman_yolov5m.pt'
     if os.path.exists(wpath):
-        
-        os.system("python track.py --yolo_weights test_detection/yolov5/weights/crowdhuman_yolov5m.pt --img 352 --save-vid --save-txt --classes 1 --conf-thres 0.4 --source " + vpath)
+        st.write('start prediction')
+        os.system("python test_detection/track.py --yolo_weights test_detection/yolov5/weights/crowdhuman_yolov5m.pt --img 352 --save-vid --save-txt --classes 1 --conf-thres 0.4 --source " + vpath)
         os.system("ffmpeg -i test_detection/inference/output/"+vname + " -vcodec libx264 -y test_detection/inference/output/output_video.mp4")
         path = 'test_detection/inference/output/output_video.mp4'
         if os.path.exists(path):
@@ -110,13 +111,14 @@ def main():
 if __name__ == '__main__':
 
   os.system('git clone --recurse-submodules https://github.com/nnassime/test_detection.git')
-  st.write(os.listdir('test_detection/'))
+  #st.write(os.listdir('test_detection/'))
 
-  os.system('python test_detection/__streamlit_app.py')
+  #os.system('python test_detection/__streamlit_app.py')
 
   load_model()
   st.write("bismillah")
   print("bismillah")
+  
   main() 
     
   st.write('out function ', os.listdir('test_detection/data/'))