Update app.py

app.py

@@ -13,26 +13,26 @@ def image_to_base64(image):
     buf = BytesIO()
     image.save(buf, "JPEG")
     buf.seek(0)
-    return f"data:image/jpeg;base64,{base64.b64encode(buf.getvalue()).decode()}"
+    return base64.b64encode(buf.getvalue()).decode("utf-8")
 
 def caption(image):
-    out = ""
-    for chunk in client.chat.completions.create(
+    image = image_to_base64(image)
+    return client.chat.completions.create(
         model="openai-large",
         messages=[
             {
                 "role": "user",
                 "content": [
-                    {"type": "image", "url": image_to_base64(image)},
+                    {
+                        "type": "image",
+                        "image_url": {"url": f"data:image/jpeg;base64,{image}"}
+                    },
                     {"type": "text", "text": PROMPT}
                ]
             }
         ],
-        max_tokens=1024,
-        stream=True
-    ):
-        out += chunk.choices[0].delta.content
-        yield out
+        max_tokens=1024
+    ).choices[0]
 
 gr.Interface(
     caption,