Update app.py

app.py

@@ -15,7 +15,7 @@ def image_to_base64(image):
     buf.seek(0)
     return base64.b64encode(buf.getvalue()).decode("utf-8")
 
-def caption(image):
+def caption(image, prompt):
     image = image_to_base64(image)
     return client.chat.completions.create(
         model="openai-large",
@@ -27,7 +27,7 @@ def caption(image):
                         "type": "image_url",
                         "image_url": {"url": f"data:image/jpeg;base64,{image}"}
                     },
-                    {"type": "text", "text": PROMPT}
+                    {"type": "text", "text": prompt}
                ]
             }
         ],
@@ -36,6 +36,10 @@ def caption(image):
 
 gr.Interface(
     caption,
-    inputs=gr.Image(type="pil", label="Image"),
-    outputs=gr.Textbox(label="Caption")
+    inputs=[
+        gr.Image(type="pil", label="Image"), 
+        gr.Textbox(label="Prompt", value=PROMPT)
+    ],
+    outputs=gr.Textbox(label="Caption"),
+    title="Image Captioning"
 ).launch(debug=True)