Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,7 @@ def process_video(video_file, question):
|
|
| 64 |
input = processor([prompt], videos=[video_clip], padding=True, return_tensors="pt").to(model.device)
|
| 65 |
|
| 66 |
# Generate output
|
| 67 |
-
generate_kwargs = {"max_new_tokens":
|
| 68 |
output = model.generate(**input, **generate_kwargs)
|
| 69 |
generated_text = processor.batch_decode(output, skip_special_tokens=True)[0]
|
| 70 |
|
|
|
|
| 64 |
input = processor([prompt], videos=[video_clip], padding=True, return_tensors="pt").to(model.device)
|
| 65 |
|
| 66 |
# Generate output
|
| 67 |
+
generate_kwargs = {"max_new_tokens": 500, "do_sample": False, "top_p": 0.9}
|
| 68 |
output = model.generate(**input, **generate_kwargs)
|
| 69 |
generated_text = processor.batch_decode(output, skip_special_tokens=True)[0]
|
| 70 |
|