Update app.py

app.py

@@ -34,11 +34,24 @@ def analyze_text(model_label, text):
     try:
         result = analyzer(text)[0]
         label = result.get("label", "UNKNOWN")
-        score = round(result.get("score", 3))
+        score = round(result.get("score", 0), 3)
+
+        # Fix model-specific labels for clarity
+        if "cardiffnlp/twitter-roberta-base-sentiment" in model_name:
+            label_map = {"LABEL_0": "Negative", "LABEL_1": "Neutral", "LABEL_2": "Positive"}
+            label = label_map.get(label, label)
+
+        elif "nlptown/bert-base-multilingual-uncased-sentiment" in model_name:
+            label = label.replace("LABEL_", "⭐️ ")  # optional, fun visual
+
+        # Strip out any extra text or URLs that sneak into outputs
+        label = str(label).split("https://")[0].strip()
+
         return f"{label} ({score})"
     except Exception as e:
         return f"Error during inference: {e}"
 
+
 # Build Gradio interface
 with gr.Blocks(title="CIS1160 – Inference Explorer") as demo:
     gr.Markdown(