ManishThota commited on
Commit
ae8423a
·
verified ·
1 Parent(s): fa7747b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def predict_answer(video, question, max_tokens=100):
67
  answers = []
68
  for i in range(len(frames)):
69
  image = extract_frames(frames[i])
70
- image_tensor = model.image_preprocess(image)
71
 
72
  # Generate the answer
73
  output_ids = model.generate(
 
67
  answers = []
68
  for i in range(len(frames)):
69
  image = extract_frames(frames[i])
70
+ image_tensor = model.image_preprocess([image])
71
 
72
  # Generate the answer
73
  output_ids = model.generate(