Update dino/app.py

dino/app.py

@@ -11,7 +11,10 @@ os.environ['KMP_DUPLICATE_LIB_OK']='True'
 
 def func(resize, video):
     output_dir = '/tmp/outputs'
-    subprocess.call(f"ffmpeg -ss 00:00:00 -i {video} -to 00:00:05 -c copy /tmp/input.mp4".split())
+    input_video = '/tmp/input.mp4'
+    if os.path.exists(input_video):
+        os.remove(input_video)
+    subprocess.call(f"ffmpeg -ss 00:00:00 -i {video} -to 00:00:05 -c copy {input_video}".split())
     if os.path.exists(output_dir):
         shutil.rmtree(output_dir)
 
@@ -22,7 +25,7 @@ def func(resize, video):
         patch_size=8,
         pretrained_weights="dino_deitsmall8_pretrain.pth",
         checkpoint_key="teacher",
-        input_path='/tmp/input.mp4',
+        input_path=input_video,
         output_path=output_dir,
         threshold=0.6,
         resize=resize,