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.gca()
+    return plt.gcf()
         
 
 demo = gr.Blocks()