Update app.py

app.py

@@ -81,8 +81,8 @@ def prediction(vname):
         with st.spinner('Running detection...'):
             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/output_'+vname
+            #st.write(os.path.exists(path), os.path.exists('inference/output/output'+vname))
             
         with st.spinner('Video loading...'):
             os.system("ffmpeg -i inference/output/"+vname + " -vcodec libx264 -y inference/output/output_"+vname)
@@ -183,7 +183,7 @@ 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/yolov5/weights/'))
+  #st.write(os.listdir('test_detection/'))
+  #st.write(os.listdir('test_detection/yolov5/weights/'))
   load_model()
   main()