Update app.py

app.py

@@ -44,7 +44,7 @@ def plot():
     cm = confusion_matrix(actual, pred)
     disp = ConfusionMatrixDisplay(confusion_matrix=cm)
     disp.plot()
-    return plt
+    return plt.gca()
         
 
 demo = gr.Blocks()
@@ -87,7 +87,7 @@ with demo:
             interface = gr.Interface(
                 plot,
                 [],
-                "image"
+                "plot"
             )
 
 demo.launch()