Update app.py

app.py

@@ -1,9 +1,17 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
+from huggingface_hub import login
+import os
 
 # Load the original pre-trained model
 def load_model(model_name):
+    hf_token = os.getenv('HUGGINGFACE_TOKEN')
+    if not hf_token:
+        return "Error: Hugging Face token not found. Please set it as a secret."
+    
+    # Login to Hugging Face Hub
+    login(hf_token)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return tokenizer, model