ManishThota commited on
Commit
634326a
1 Parent(s): 06ae73a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def predict_answer(image, video, question, max_tokens=100):
40
  # Process as an image
41
  image = image.convert("RGB")
42
  input_ids = tokenizer(question, return_tensors='pt').input_ids.to(device)
43
- image_tensor = model.image_preprocess(frame)
44
 
45
  #Generate the answer
46
  output_ids = model.generate(
 
40
  # Process as an image
41
  image = image.convert("RGB")
42
  input_ids = tokenizer(question, return_tensors='pt').input_ids.to(device)
43
+ image_tensor = model.image_preprocess(image)
44
 
45
  #Generate the answer
46
  output_ids = model.generate(