Upload app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import torch.nn.functional as F
+from huggingface_hub import login
+import os
+
+# Authenticate with Hugging Face using token from environment variable
+login(os.environ.get("HUGGINGFACE_TOKEN"))
 
 # Load MentalBERT model & tokenizer
 MODEL_NAME = "mental/mental-bert-base-uncased"
@@ -41,11 +46,12 @@ iface = gr.Interface(
     inputs=gr.Textbox(label="Enter text to analyze", lines=3),
     outputs=gr.Json(label="Emotion Analysis"),
     title="MentalBERT Emotion Analysis",
-    description="Analyze the emotional content of text using MentalBERT",
+    description="Analyze the emotional content of text using MentalBERT (specialized for mental health content)",
     examples=[
-        ["I feel really happy today!"],
-        ["I'm feeling quite stressed and overwhelmed"],
-        ["The weather is nice outside"]
+        ["I feel really anxious about my upcoming presentation"],
+        ["I've been feeling quite depressed lately"],
+        ["I'm managing my stress levels well today"],
+        ["Just had a great therapy session!"]
     ]
 )