Geraldine commited on
Commit
291a8ed
·
verified ·
1 Parent(s): 7159d2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def describe_image(image_array):
46
  # Excepted output: '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>Describe this image.<|im_end|>\n<|im_start|>assistant\n'
47
 
48
  inputs = processor(
49
- text=[text_prompt], images=[img_str], padding=True, return_tensors="pt"
50
  )
51
  inputs = inputs.to("cpu")
52
 
 
46
  # Excepted output: '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>Describe this image.<|im_end|>\n<|im_start|>assistant\n'
47
 
48
  inputs = processor(
49
+ text=[text_prompt], images=[f"data:image/png;base64,{img_str}"], padding=True, return_tensors="pt"
50
  )
51
  inputs = inputs.to("cpu")
52