Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -8,7 +8,7 @@ from data_cleaning_env import DataCleaningEnvironment, CleaningAction
|
|
| 8 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 9 |
API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 10 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 11 |
-
MODEL_NAME = os.getenv("MODEL_NAME", "
|
| 12 |
if HF_TOKEN is None:
|
| 13 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 14 |
|
|
|
|
| 8 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 9 |
API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 10 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 11 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "meta-llama/Llama-3.1-8B-Instruct:cerebras")
|
| 12 |
if HF_TOKEN is None:
|
| 13 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 14 |
|