Spaces:
Sleeping
Sleeping
Joash
commited on
Commit
·
2f11a3f
1
Parent(s):
8864ec7
Update app with proper token handling
Browse files
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,
|