ManishThota commited on
Commit
9309155
1 Parent(s): 858540e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def predict_answer(video, image, question):
85
  elif video:
86
  frames = video_to_frames(video)
87
  image = extract_frames(frames[2])
88
- image_tensor = model.image_preprocess(image)
89
  # Generate the answer
90
  output_ids = model.generate(
91
  input_ids,
 
85
  elif video:
86
  frames = video_to_frames(video)
87
  image = extract_frames(frames[2])
88
+ image_tensor = model.image_preprocess([image])
89
  # Generate the answer
90
  output_ids = model.generate(
91
  input_ids,