Joash commited on
Commit
2f11a3f
·
1 Parent(s): 8864ec7

Update app with proper token handling

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,9 +18,12 @@ logging.basicConfig(level=logging.INFO)
18
  logger = logging.getLogger(__name__)
19
 
20
  # Environment variables
21
- HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
22
  MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2b-it")
23
 
 
 
 
24
  # Create data directory for persistence
25
  DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
26
  os.makedirs(DATA_DIR, exist_ok=True)
@@ -110,9 +113,6 @@ class CodeReviewer:
110
  def initialize_model(self):
111
  """Initialize the model and tokenizer."""
112
  try:
113
- if HF_TOKEN:
114
- login(token=HF_TOKEN, add_to_git_credential=False)
115
-
116
  logger.info("Loading tokenizer...")
117
  self.tokenizer = AutoTokenizer.from_pretrained(
118
  MODEL_NAME,
 
18
  logger = logging.getLogger(__name__)
19
 
20
  # Environment variables
21
+ HF_TOKEN = os.getenv("HUGGING_FACE_TOKEN", "v4e-o3_7lUO4cn8T9m9p5GNSI") # Default to the token if not set in Space
22
  MODEL_NAME = os.getenv("MODEL_NAME", "google/gemma-2b-it")
23
 
24
+ # Login to Hugging Face
25
+ login(token=HF_TOKEN)
26
+
27
  # Create data directory for persistence
28
  DATA_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
29
  os.makedirs(DATA_DIR, exist_ok=True)
 
113
  def initialize_model(self):
114
  """Initialize the model and tokenizer."""
115
  try:
 
 
 
116
  logger.info("Loading tokenizer...")
117
  self.tokenizer = AutoTokenizer.from_pretrained(
118
  MODEL_NAME,