ManishThota commited on
Commit
3295429
1 Parent(s): 0b1b270

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def predict_answer(image, video, question, max_tokens=100):
83
  frames = video_to_frames(video)
84
  answers = []
85
  for frame in frames:
86
- image = Image.open(extract_frames(frame)).convert("RGB")
87
  image_tensor = model.image_preprocess(image)
88
 
89
  # Generate the answer
 
83
  frames = video_to_frames(video)
84
  answers = []
85
  for frame in frames:
86
+ image = extract_frames(frame)
87
  image_tensor = model.image_preprocess(image)
88
 
89
  # Generate the answer