Update app.py

app.py

@@ -34,19 +34,21 @@ def describe_image(image_array):
         {
             "role": "user",
             "content": [
-                {"type": "image", "image": f"data:image/png;base64,{image}"},
+                {
+                    "type": "image",
+                },
                 {"type": "text", "text": "Make a very detailed description of the image."},
             ],
         }
     ]
 
-    text_prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
+    text_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
     # 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'
 
     inputs = processor(
-        text=[text_prompt], images=[image], padding=True, return_tensors="pt"
+        text=[text_prompt], images=[image_str], padding=True, return_tensors="pt"
     )
-    inputs = inputs.to("cuda")
+    inputs = inputs.to("cpu")
 
     # Inference: Generation of the output
     output_ids = model.generate(**inputs, max_new_tokens=128)