Update app.py
Browse files
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 |
|