Update app.py

app.py

@@ -3,37 +3,33 @@ import os
 from io import BytesIO
 
 import gradio as gr
-import requests
+from huggingface_hub import InferenceClient
 
-API_URL = "https://text.pollinations.ai/openai"
 PROMPT = os.environ.get("PROMPT", "Describe this image.")
 
+client = InferenceClient(model="https://text.pollinations.ai/openai")
+
 def image_to_base64(image):
     buf = BytesIO()
     image.save(buf, "JPEG")
-    return base64.b64encode(buf.getvalue()).decode("utf-8")
+    buf.seek(0)
+    return f"data:image/jpeg;base64,{base64.b64encode(buf.getvalue()).decode()}"
 
 def caption(image):
-    data = {
-        "model": "openai-large",
-        "messages": [
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": PROMPT},
-                    {
-                        "type": "image_url",
-                        "image_url": {"url": f"data:image/jpeg;base64,{image_to_base64(image)}"}
-                    }
-                ]
-            }
-        ],
-        "private": "true"
-    }
-    response = requests.post(API_URL, json=data)
-    if response.ok:
-        return response.json()["choices"][0]["message"]["content"]
-    return ""
+    out = ""
+    for chunk in client.chat.completions.create(
+        model="openai-large",
+        messages=[{
+            "role": "user",
+            "content": [
+                {"type": "text", "text": PROMPT},
+                {"type": "image_url", "image_url": {"url": image_to_base64(image)}}
+            ]
+        }],
+        stream=True
+    ):
+        out += chunk.choices[0].delta.content
+        yield out
 
 gr.Interface(
     caption,