Update app.py

app.py

@@ -6,6 +6,7 @@ import torch.nn.functional as F
 from typing import Literal, Any
 import gradio as gr
 import spaces
+from io import BytesIO
 
 
 class Classifier:
@@ -74,7 +75,14 @@ def draw_bar_chart(data: dict[str, list[str | float]]):
 
     plt.tight_layout()
 
-    return plt
+    bimg = BytesIO()
+    plt.save(img, format="png")
+    plt.close
+
+    bimg.seek(0)
+    img = Image.open(bimg)
+
+    return img
 
 
 def get_layout():
@@ -137,7 +145,7 @@ def get_layout():
     
             with gr.Row(equal_height=True):
                 image_input = gr.Image(label="上傳影像", type="pil")
-                bar_chart = gr.Plot(label="分類結果")
+                chart = gr.Image(label="分類結果")
 
             start_button = gr.Button("開始分類", variant="primary")
             gr.HTML(
@@ -146,7 +154,7 @@ def get_layout():
             start_button.click(
                 fn=Classifier().predict,
                 inputs=image_input,
-                outputs=bar_chart,
+                outputs=chart,
             )
 
     return demo