Fix token handling with proper git credentials

app.py

@@ -9,6 +9,10 @@ import json
 from typing import List, Dict
 import warnings
 import spaces
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Filter out warnings
 warnings.filterwarnings('ignore')
@@ -18,12 +22,9 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Environment variables
-HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN", "v4e-o3_7lUO4cn8T9m9p5GNSI")  # Default to the token if not set in Space
+HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
 MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2b-it")
 
-# Login to Hugging Face
-login(token=HF_TOKEN)
-
 # Create data directory for persistence
 DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
 os.makedirs(DATA_DIR, exist_ok=True)
@@ -113,6 +114,15 @@ class CodeReviewer:
     def initialize_model(self):
         """Initialize the model and tokenizer."""
         try:
+            # Login to Hugging Face with git credential
+            if HF_TOKEN:
+                try:
+                    login(token=HF_TOKEN, add_to_git_credential=True)
+                    logger.info("Successfully logged in to Hugging Face")
+                except Exception as e:
+                    logger.error(f"Error logging in to Hugging Face: {e}")
+                    return False
+            
             logger.info("Loading tokenizer...")
             self.tokenizer = AutoTokenizer.from_pretrained(
                 MODEL_NAME,