Update app.py

app.py

@@ -89,10 +89,10 @@ if uploaded_file is not None and analyze_button:
                 "content": [
                     {"type": "text", "text": prompt_text},
                     {
-                        "type": "image",
-                        "image": {
-                            # Provide the image bytes directly
-                            "bytes": base64.b64decode(base64_image)
+                        "type": "image_url",
+                        "image_url": {
+                            # Since we have the image in base64, we need to use a data URL
+                            "url": f"data:image/jpeg;base64,{base64_image}"
                         },
                     },
                 ],
@@ -105,11 +105,11 @@ if uploaded_file is not None and analyze_button:
             completion = client.chat.completions.create(
                 model="meta-llama/Llama-3.2-11B-Vision-Instruct",
                 messages=messages,
-                max_tokens=1200
+                max_tokens=500  # Adjusted to match the template
             )
 
             # Extract the assistant's response
-            assistant_response = completion.get("choices")[0]["message"]["content"]
+            assistant_response = completion.choices[0].message['content']
 
             # Display the response
             st.markdown(assistant_response)