ahmed-masry commited on
Commit
7bd1a65
1 Parent(s): b6a5eae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -21,6 +21,8 @@ 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=input_prompt, images=image, return_tensors="pt")
25
  inputs = {k: v.to(device) for k, v in inputs.items()}
26
 
 
21
 
22
  input_prompt = f"<image>\n Question: {input_text} Answer: "
23
  image = image.convert("RGB")
24
+ print(input_text)
25
+ print(image.size)
26
  inputs = processor(text=input_prompt, images=image, return_tensors="pt")
27
  inputs = {k: v.to(device) for k, v in inputs.items()}
28