arjunanand13 commited on
Commit
e12f44e
·
verified ·
1 Parent(s): 36f26b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -28,8 +28,9 @@ def analyze_image(image, prompt):
28
  ]
29
  input_text = processor.apply_chat_template(messages, add_generation_prompt=True)
30
  inputs = processor(
31
- images=image,
32
- text=input_text,
 
33
  return_tensors="pt"
34
  ).to(model.device)
35
  with torch.no_grad():
 
28
  ]
29
  input_text = processor.apply_chat_template(messages, add_generation_prompt=True)
30
  inputs = processor(
31
+ image,
32
+ input_text,
33
+ add_special_tokens=False,
34
  return_tensors="pt"
35
  ).to(model.device)
36
  with torch.no_grad():