Update app.py

app.py

@@ -1,7 +1,16 @@
 import gradio as gr
 import torch
 import random
+import os
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from huggingface_hub import login
+
+# Login to Hugging Face using token from environment variable
+hf_token = os.getenv("HUGGINGFACE_TOKEN")
+if hf_token:
+    login(token=hf_token)
+else:
+    raise ValueError("HUGGINGFACE_TOKEN environment variable not set.")
 
 # Load the trained model and tokenizer
 model_name = "your-username/hate-speech-classifier"  # Replace with your actual Hugging Face model repo