Update app.py

app.py

@@ -2,22 +2,30 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import logging
-from huggingface_hub import login, HfApi
+from huggingface_hub import login
 from datetime import datetime
 import pandas as pd
 from datasets import Dataset, load_dataset
+import os
 
 logging.basicConfig(level=logging.INFO)
 
 # ==============================================
-#  CONFIGURATION – ONLY CHANGE THESE TWO LINES
+#  CONFIGURATION – READ FROM SECRETS
 # ==============================================
 HF_DATASET_NAME = "AdhamAshraf/slanggpt-feedback-dataset"   # your feedback dataset
-HF_TOKEN = "hf_your_token_here"                             # your write token
 MODEL_NAME = "AdhamAshraf/SlangGPT"                         # your generation model
+HF_TOKEN = os.getenv("HF_TOKEN")                            # read from Space secret
 
-# Login to Hugging Face
-login(token=HF_TOKEN, add_to_git_credential=True)
+if not HF_TOKEN:
+    raise RuntimeError(
+        "HF_TOKEN environment variable not set. "
+        "Please add a secret named 'HF_TOKEN' with your Hugging Face write token."
+    )
+
+# Login to Hugging Face (without git credential helper – avoids 401 in Spaces)
+login(token=HF_TOKEN)
+print("✅ Logged in to Hugging Face Hub")
 
 # ==============================================
 #  LOAD GENERATION MODEL