Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- inference.py +3 -3
inference.py
CHANGED
|
@@ -6,9 +6,9 @@ from openai import OpenAI
|
|
| 6 |
from data_cleaning_env import DataCleaningEnvironment, CleaningAction
|
| 7 |
|
| 8 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 9 |
-
API_BASE_URL = os.getenv("API_BASE_URL", "https://
|
| 10 |
-
|
| 11 |
-
|
| 12 |
if HF_TOKEN is None:
|
| 13 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 14 |
|
|
|
|
| 6 |
from data_cleaning_env import DataCleaningEnvironment, CleaningAction
|
| 7 |
|
| 8 |
# ββ Config ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 9 |
+
API_BASE_URL = os.getenv("API_BASE_URL", "https://api.groq.com/openai/v1")
|
| 10 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "openai/gpt-oss-120b") # Groq model name
|
| 11 |
+
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 12 |
if HF_TOKEN is None:
|
| 13 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 14 |
|