prettier output

app.py

@@ -18,9 +18,13 @@ tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
 
 
 def classify(text):
-    return reverse_map[
-        model(**tokenizer(text, return_tensors="pt")).logits.argmax(-1).item()
-    ]
+    return (
+        reverse_map[
+            model(**tokenizer(text, return_tensors="pt")).logits.argmax(-1).item()
+        ]
+        .replace("_", " ")
+        .capitalize()
+    )
 
 
 iface = gr.Interface(fn=classify, inputs="text", outputs="text")