ahmed-masry commited on
Commit
6d0a225
·
verified ·
1 Parent(s): bbee3f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def predict(image, input_text):
21
 
22
  input_prompt = f"<image>\n Question: {input_text} Answer: "
23
  image = image.convert("RGB")
24
- inputs = processor(text=prompt, images=image, return_tensors="pt")
25
  inputs = {k: v.to(device) for k, v in inputs.items()}
26
 
27
  # change type if pixel_values in inputs to fp16.
 
21
 
22
  input_prompt = f"<image>\n Question: {input_text} Answer: "
23
  image = image.convert("RGB")
24
+ inputs = processor(text=input_prompt, images=image, return_tensors="pt")
25
  inputs = {k: v.to(device) for k, v in inputs.items()}
26
 
27
  # change type if pixel_values in inputs to fp16.