Update app.py

app.py

@@ -9,11 +9,11 @@ import re
 
 # 将图像转换为 base64，以便在 gradio 中显示
 def get_image_data(fig):
-    buf = io.BytesIO()
-    fig.savefig(buf, format='PNG')
-    buf.seek(0)
-    img = Image.open(buf)
-    return img
+    img_byte_arr = io.BytesIO() 
+    img.save(img_byte_arr, format='PNG')
+    img_byte_arr = img_byte_arr.getvalue()
+    img_str = base64.b64encode(img_byte_arr).decode('utf-8')
+    return img_str
 
 # 执行 Python 代码并生成图像
 def execute_code(code):
@@ -71,7 +71,7 @@ def gpt_inference(base_url, model, openai_key, prompt):
 iface = gr.Interface(
     fn=gpt_inference, 
     inputs=["text", gr.inputs.Dropdown(choices=["gpt3.5-turbo", "gpt4"], label="Model"), "text", "text"], 
-    outputs=gr.outputs.Image(type="pil"),
+    outputs=gr.Image(),
     input_labels=["Base URL", "Model", "OpenAI Key","Prompt"]
 )
 iface.launch()