ManishThota commited on
Commit
be4f38b
·
verified ·
1 Parent(s): 69f97c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": 100, "do_sample": True, "top_p": 0.9}
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