Spaces:
Running
on
Zero
Running
on
Zero
chenlin
commited on
Commit
·
4dcb992
1
Parent(s):
49bae8f
fix bug
Browse files
app.py
CHANGED
@@ -272,7 +272,7 @@ def generate_slidingcaptioning(video_path):
|
|
272 |
def generate_fastcaptioning(video_path):
|
273 |
q = 'Here are a few key frames of a video, discribe this video in detail.'
|
274 |
query = f'[UNUSED_TOKEN_146]user\n{q}[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n'
|
275 |
-
imgs = load_quota_video(video_path
|
276 |
img = img_process(imgs)
|
277 |
with torch.cuda.amp.autocast():
|
278 |
response = model_gen(model, query, img, hd_num=16,
|
|
|
272 |
def generate_fastcaptioning(video_path):
|
273 |
q = 'Here are a few key frames of a video, discribe this video in detail.'
|
274 |
query = f'[UNUSED_TOKEN_146]user\n{q}[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n'
|
275 |
+
imgs = load_quota_video(video_path)
|
276 |
img = img_process(imgs)
|
277 |
with torch.cuda.amp.autocast():
|
278 |
response = model_gen(model, query, img, hd_num=16,
|