Update minigpt4_video_demo.py

minigpt4_video_demo.py

@@ -17,10 +17,19 @@ from PIL import Image
 from moviepy.editor import VideoFileClip
 from theme import minigptlv_style, custom_css,text_css
 
-from huggingface_hub import login
+from huggingface_hub import login, hf_hub_download
 hf_token = os.environ.get('HF_TKN')
 login(token=hf_token)
 
+hf_hub_download(
+    repo_id='Vision-CAIR/MiniGPT4-Video',
+    filename='video_llama_checkpoint_last.pth',
+    local_dir='checkpoints',
+    local_dir_use_symlinks=False,
+)
+
+import spaces
+
 def create_video_grid(images, rows, cols,save_path):
     image_width, image_height = images[0].size
     grid_width = cols * image_width
@@ -132,7 +141,7 @@ def generate_subtitles(video_path):
         print("error",video_path)
         return None
     
-
+@spaces.GPU()
 def run (video_path,instruction,model,vis_processor,gen_subtitles=True):
     if gen_subtitles:
         subtitle_path=generate_subtitles(video_path)