Update app.py

app.py

@@ -79,9 +79,7 @@ def predict2(image_np):
     
     return result_pil_img
 def write_video(video_in_filepath, video_out_filepath, detection_model):
-    if not os.path.exists(video_in_filepath):
-        print('video filepath not valid')
-    
+       
     video_reader = cv2.VideoCapture(video_in_filepath)
     
     nb_frames = int(video_reader.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -117,6 +115,11 @@ def write_video(video_in_filepath, video_out_filepath, detection_model):
     video_writer.release() 
     cv2.destroyAllWindows() 
     cv2.waitKey(1)
+    
+def predict_video (video_file_name):
+    detected_video_file = "detected_video.mp4"
+    write_video(video_file_name,detected_video_file,detection_model)
+    return detected_video_file
 
 REPO_ID = "YEHTUT/tfodmodel"
 detection_model = load_model()
@@ -129,7 +132,7 @@ Image_tab = Interface(fn=predict,
              inputs=gr.Image(type="pil"),
              outputs=gr.Image(type="pil")
              )
-Video_tab = Interface(fn=predict,
+Video_tab = Interface(fn=predict_video,
              inputs=gr.Video,
              outputs=gr.Video
              )