Spaces:
Running
on
Zero
Running
on
Zero
ManishThota
commited on
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 |
|