Upload 2 files

app.py

@@ -6,16 +6,27 @@ from huggingface_hub import login
 import os
 
 # Authenticate with Hugging Face using token from environment variable
-login(os.environ.get("HUGGINGFACE_TOKEN"))
+try:
+    hf_token = os.environ.get("HUGGINGFACE_TOKEN")
+    if hf_token:
+        login(hf_token)
+    else:
+        print("Warning: HUGGINGFACE_TOKEN not found in environment variables")
+except Exception as e:
+    print(f"Authentication error: {e}")
 
 # Load MentalBERT model & tokenizer
-MODEL_NAME = "mental/mental-bert-base-uncased"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForSequenceClassification.from_pretrained(
-    MODEL_NAME,
-    num_labels=2,
-    problem_type="single_label_classification"
-)
+try:
+    MODEL_NAME = "mental/mental-bert-base-uncased"
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+    model = AutoModelForSequenceClassification.from_pretrained(
+        MODEL_NAME,
+        num_labels=2,
+        problem_type="single_label_classification"
+    )
+except Exception as e:
+    print(f"Error loading model: {e}")
+    raise
 
 LABELS = {
     "neutral": {"index": 0, "description": "Emotionally balanced or calm"},
@@ -52,8 +63,9 @@ iface = gr.Interface(
         ["I've been feeling quite depressed lately"],
         ["I'm managing my stress levels well today"],
         ["Just had a great therapy session!"]
-    ]
+    ],
+    allow_flagging="never"
 )
 
-# Launch the interface
-iface.launch()
+# Launch the interface with CORS support
+iface.launch(share=True, server_name="0.0.0.0")

requirements.txt

@@ -1,5 +1,6 @@
 gradio==4.13.0
 transformers==4.35.2
 torch==2.1.1
-uvicorn==0.24.0
+huggingface-hub==0.19.4
+numpy==1.26.2
 pydantic==2.5.2