Spaces:
Sleeping
Sleeping
Joash
commited on
Commit
·
2130931
1
Parent(s):
9f46175
Remove hardcoded token and require it from environment variables
Browse files- src/config.py +3 -2
src/config.py
CHANGED
@@ -26,7 +26,8 @@ class Config:
|
|
26 |
MAX_OUTPUT_LENGTH = int(os.getenv("MAX_OUTPUT_LENGTH", 1024))
|
27 |
TEMPERATURE = float(os.getenv("TEMPERATURE", 0.7))
|
28 |
TOP_P = float(os.getenv("TOP_P", 0.95))
|
29 |
-
|
|
|
30 |
|
31 |
# Database Settings
|
32 |
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./code_review.db")
|
@@ -59,7 +60,7 @@ class Config:
|
|
59 |
logger.info(f"DEBUG: {Config.DEBUG}")
|
60 |
|
61 |
if not Config.HUGGING_FACE_TOKEN:
|
62 |
-
raise ValueError("HUGGING_FACE_TOKEN must be set to access Hugging Face
|
63 |
if Config.TEMPERATURE < 0 or Config.TEMPERATURE > 1:
|
64 |
raise ValueError("TEMPERATURE must be between 0 and 1.")
|
65 |
if Config.TOP_P < 0 or Config.TOP_P > 1:
|
|
|
26 |
MAX_OUTPUT_LENGTH = int(os.getenv("MAX_OUTPUT_LENGTH", 1024))
|
27 |
TEMPERATURE = float(os.getenv("TEMPERATURE", 0.7))
|
28 |
TOP_P = float(os.getenv("TOP_P", 0.95))
|
29 |
+
# Token must be set in environment variables
|
30 |
+
HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
|
31 |
|
32 |
# Database Settings
|
33 |
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./code_review.db")
|
|
|
60 |
logger.info(f"DEBUG: {Config.DEBUG}")
|
61 |
|
62 |
if not Config.HUGGING_FACE_TOKEN:
|
63 |
+
raise ValueError("HUGGING_FACE_TOKEN must be set in environment variables to access Hugging Face models.")
|
64 |
if Config.TEMPERATURE < 0 or Config.TEMPERATURE > 1:
|
65 |
raise ValueError("TEMPERATURE must be between 0 and 1.")
|
66 |
if Config.TOP_P < 0 or Config.TOP_P > 1:
|