Update app.py

app.py

@@ -33,7 +33,7 @@ def load_model():
 
 # Ft saving uploaded video to directory
 def save_uploaded_vid(uploadedfile):
-    with open(os.path.join("data", uploadedfile.name),"wb") as f:
+    with open(os.path.join("test_detection/data", uploadedfile.name),"wb") as f:
         f.write(uploadedfile.getbuffer())
     return st.success("Video saved in data dir ")
 
@@ -89,8 +89,8 @@ def prediction(vname):
         st.write('start prediction')
         st.write(os.listdir('test_detection/deep_sort_pytorch/configs/'))
         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 /inference/output/"+vname + " -vcodec libx264 -y test_detection/inference/output/output_video.mp4")
-        path = 'test_detection/inference/output/output_video.mp4'
+        os.system("ffmpeg -i inference/output/"+vname + " -vcodec libx264 -y inference/output/output_video.mp4")
+        path = 'inference/output/output_video.mp4'
         if os.path.exists(path):
             video_file = open(path, 'rb')
             video_bytes = video_file.read()