Spaces:
Sleeping
Sleeping
Update minigpt4_video_demo.py
Browse files- minigpt4_video_demo.py +3 -1
minigpt4_video_demo.py
CHANGED
@@ -147,7 +147,7 @@ def generate_subtitles(video_path):
|
|
147 |
print("error",video_path)
|
148 |
return None
|
149 |
|
150 |
-
|
151 |
def run (video_path,instruction,model,vis_processor,gen_subtitles=True):
|
152 |
if gen_subtitles:
|
153 |
subtitle_path=generate_subtitles(video_path)
|
@@ -245,10 +245,12 @@ existed_subtitles={}
|
|
245 |
for sub in os.listdir(inference_subtitles_folder):
|
246 |
existed_subtitles[sub.split('.')[0]]=True
|
247 |
|
|
|
248 |
def gradio_demo_local(video_path,has_sub,instruction):
|
249 |
pred=run(video_path,instruction,model,vis_processor,gen_subtitles=has_sub)
|
250 |
return pred
|
251 |
|
|
|
252 |
def gradio_demo_youtube(youtube_url,has_sub,instruction):
|
253 |
video_path=get_video_url(youtube_url,has_sub)
|
254 |
pred=run(video_path,instruction,model,vis_processor,gen_subtitles=has_sub)
|
|
|
147 |
print("error",video_path)
|
148 |
return None
|
149 |
|
150 |
+
|
151 |
def run (video_path,instruction,model,vis_processor,gen_subtitles=True):
|
152 |
if gen_subtitles:
|
153 |
subtitle_path=generate_subtitles(video_path)
|
|
|
245 |
for sub in os.listdir(inference_subtitles_folder):
|
246 |
existed_subtitles[sub.split('.')[0]]=True
|
247 |
|
248 |
+
@spaces.GPU()
|
249 |
def gradio_demo_local(video_path,has_sub,instruction):
|
250 |
pred=run(video_path,instruction,model,vis_processor,gen_subtitles=has_sub)
|
251 |
return pred
|
252 |
|
253 |
+
@spaces.GPU()
|
254 |
def gradio_demo_youtube(youtube_url,has_sub,instruction):
|
255 |
video_path=get_video_url(youtube_url,has_sub)
|
256 |
pred=run(video_path,instruction,model,vis_processor,gen_subtitles=has_sub)
|