Create app.py

app.py

@@ -0,0 +1,42 @@
+import base64
+import os
+from io import BytesIO
+
+import gradio as gr
+import requests
+
+API_URL = "https://text.pollinations.ai/openai"
+PROMPT = os.environ.get("PROMPT", "Describe this image.")
+
+def image_to_base64(image):
+    buf = BytesIO()
+    image.save(buf, "JPEG")
+    return base64.b64encode(buf.getvalue()).decode("utf-8")
+
+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 ""
+
+gr.Interface(
+    caption,
+    inputs=gr.Image(type="pil", label="Image"),
+    outputs=gr.Textbox(label="Caption")
+).launch(debug=True)