Spaces:
Sleeping
Sleeping
Updated API
Browse files- inference.py +4 -4
inference.py
CHANGED
|
@@ -29,10 +29,10 @@ load_dotenv()
|
|
| 29 |
# Configuration
|
| 30 |
# ---------------------------------------------------------------------------
|
| 31 |
|
| 32 |
-
API_KEY = os.
|
| 33 |
-
API_BASE_URL = os.
|
| 34 |
-
MODEL_NAME = os.
|
| 35 |
-
ENV_URL = os.
|
| 36 |
BENCHMARK = "xsecure"
|
| 37 |
MAX_STEPS = 20
|
| 38 |
|
|
|
|
| 29 |
# Configuration
|
| 30 |
# ---------------------------------------------------------------------------
|
| 31 |
|
| 32 |
+
API_KEY = os.getenv("API_KEY", "")
|
| 33 |
+
API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 34 |
+
MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
|
| 35 |
+
ENV_URL = os.getenv("ENV_URL", "http://localhost:7860")
|
| 36 |
BENCHMARK = "xsecure"
|
| 37 |
MAX_STEPS = 20
|
| 38 |
|