Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 from joblib import load
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch.nn.functional as F
-from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
+from sklearn.metrics import accuracy_score, confusion_matrix, ConfusionMatrixDisplay
 import matplotlib.pyplot as plt
 
 # global variables to load models
@@ -44,7 +44,7 @@ def plot():
     cm = confusion_matrix(actual, pred)
     disp = ConfusionMatrixDisplay(confusion_matrix=cm)
     disp.plot()
-    return plt.gcf()
+    return (plt.gcf(), f"Accuracy: {accuracy_score(actual, pred)}")
         
 
 demo = gr.Blocks()
@@ -87,7 +87,7 @@ with demo:
             interface = gr.Interface(
                 plot,
                 [],
-                "plot"
+                ["plot", "text"]
             )
 
 demo.launch()