Update app.py

app.py

@@ -32,9 +32,11 @@ def infer(video_in):
     # path_to_video
 
     print(f"VIDEO IN PATH: {video_in}")
+    # Get the directory name
+    folder_path = os.path.dirname(video_in)
 
     # Execute the inference command
-    command = ['python', 'inference_from_video.py', '--original_args', 'ckpt/vta-ldm-clip4clip-v-large/summary.jsonl', '--model', 'ckpt/vta-ldm-clip4clip-v-large/pytorch_model_2.bin', '--data_path', video_in]
+    command = ['python', 'inference_from_video.py', '--original_args', 'ckpt/vta-ldm-clip4clip-v-large/summary.jsonl', '--model', 'ckpt/vta-ldm-clip4clip-v-large/pytorch_model_2.bin', '--data_path', folder_path]
     process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)
 
     # Create threads to handle stdout and stderr