Update app.py

app.py

@@ -2,18 +2,21 @@ import gradio as gr
 from PIL import Image
 import requests
 import base64
+import io
 import openai
 openai.api_key = ""
 openai.api_base = "https://api.deepinfra.com/v1/openai"
 def todataimage(file, ext):
-    image_data_uri = "data:image/png;base64," + base64.b64encode(file)
-    return image_data_uri
+    buffered = io.BytesIO()
+    file.save(buffered, format=ext)
+    return "data:image/png;base64,"+base64.b64encode(buffered.getvalue()).decode("utf-8")
 def caption(file, ext):
+  datimg = todataimage(file, ext)
   response = requests.post("https://russellc-comparing-captioning-models.hf.space/run/predict", json={
     "data": [
-      todataimage(file, ext),
+      datimg,
   ]}).json()
-
+  print(response)
   data = response["data"]
   chat_completion = openai.ChatCompletion.create(
     model="meta-llama/Llama-2-70b-chat-hf",
@@ -24,5 +27,5 @@ def caption(file, ext):
 def image_predict(image):
     return caption(image, "png")
 
-iface = gr.Interface(fn=image_predict, inputs="image", outputs="label")
+iface = gr.Interface(image_predict, inputs=gr.Image(type="pil"), outputs="label", flagging_options=[])
 iface.launch()