ManishThota commited on
Commit
22c70be
·
verified ·
1 Parent(s): 03269c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def predict_answer(image, question):
12
  image = image.convert("RGB")
13
 
14
  # Prepare inputs
15
- encoding = processor(image, question, return_tensors="pt").to("cuda:0", torch.float16)
16
 
17
  out = model.generate(**encoding)
18
  generated_text = processor.decode(out[0], skip_special_tokens=True)
 
12
  image = image.convert("RGB")
13
 
14
  # Prepare inputs
15
+ encoding = processor(image, question, return_tensors="pt")
16
 
17
  out = model.generate(**encoding)
18
  generated_text = processor.decode(out[0], skip_special_tokens=True)