Update app.py

app.py

@@ -79,13 +79,18 @@ def prediction(vname):
     wpath = 'test_detection/yolov5/weights/crowdhuman_yolov5m.pt'
     if os.path.exists(wpath):
         with st.spinner('Running detection...'):
+        
+            st.write('vpath befor calling track : ', vpath)
+            
             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)
             
-            path = 'inference/output/output_'+vname
-            st.write(os.path.exists(path), os.path.exists('inference/output/output'+vname))
+            path = 'inference/output/'
+            #st.write(os.path.exists(path), os.path.exists('inference/output/output'+vname))
+            
+            
+            st.write(os.listdir('test_detection/'))
             
             st.write(os.listdir(path))
-            st.write(os.listdir('inference/output/output'+vname))
             
         with st.spinner('Video loading...'):
             os.system("ffmpeg -i inference/output/"+vname + " -vcodec libx264 -y inference/output/output_"+vname)