Update app.py

app.py

@@ -17,8 +17,7 @@ def image_to_base64(image):
 
 def caption(image, prompt):
     image = image_to_base64(image)
-    out = ""
-    for chunk in client.chat.completions.create(
+    return client.chat.completions.create(
         model="openai-large",
         messages=[
             {
@@ -32,11 +31,8 @@ def caption(image, prompt):
                ]
             }
         ],
-        max_tokens=1024,
-        stream=True
-    ):
-        out += chunk.choices[0].delta.content
-        yield out
+        max_tokens=1024
+    ).choices[0].message.content
 
 gr.Interface(
     caption,