Update app.py

app.py

@@ -75,17 +75,12 @@ def draw_bar_chart(data: dict[str, list[str | float]]):
 
     plt.tight_layout()
 
-    bimg = BytesIO()
-    plt.save(img, format="png")
-    plt.close
-
-    bimg.seek(0)
-    img = Image.open(bimg)
-
-    return img
+    return plt
 
 
 def get_layout():
+    demo = gr.Interface(fn=Classifier().predict, inputs="image", outputs="plot")
+    return demo
     css = """
     .main-title {
         font-size: 24px;