Change label to bar plot

app.py

@@ -6,7 +6,7 @@ from transformers import (
 )
 from typing import Dict
 import os
-from custom_label import CustomLabel
+import pandas as pd
 from huggingface_hub import login
 login(token=os.getenv("HUGGINGFACE_TOKEN"))
 
@@ -31,14 +31,15 @@ def classify_text(text: str) -> Dict[str, float]:
                        padding=True,
                        truncation=True,
                        return_tensors='pt').to(DEVICE)
-    scores = {}
+    scores = []
     for foundation in FOUNDATIONS:
         model = MODELS[foundation]
         outputs = model(**inputs)
         outputs = torch.softmax(outputs.logits, dim=1)
         outputs = outputs[:, 1]
         score = outputs.detach().cpu().numpy()[0]
-        scores[foundation.capitalize()] = score
+        scores.append([foundation.capitalize, score])
+    scores = pd.DataFrame(scores, columns=["foundation", "score"])
     return scores
 
 
@@ -56,12 +57,13 @@ demo = gr.Interface(
         ),
     ],
     outputs=[
-        CustomLabel(
-            label="Moral foundations scores",
-            container=False,
-            show_label=False,
-            scale=10,
-            lines=10,
+        gr.BarPlot(
+            x="foundation",
+            y="score",
+            title="Moral foundations scores",
+            x_title="Moral foundation",
+            y_title="Score",
+            vertical=True,
         )
     ],
 )

requirements.txt

@@ -4,3 +4,4 @@ scipy==1.11.1
 sentencepiece==0.1.99
 torch==2.0.1
 transformers==4.31.0
+pandas==1.5.3