Merge pull request #1 from soumiguria/soumi
Browse files- inference.py +13 -4
inference.py
CHANGED
|
@@ -29,12 +29,21 @@ def post_json(url: str, payload: dict) -> dict:
|
|
| 29 |
raise Exception(f"HTTP Error {e.code}: {e.read().decode('utf-8')}")
|
| 30 |
|
| 31 |
# ββ Environment variables ββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 32 |
-
API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 33 |
-
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
-
API_KEY = HF_TOKEN or os.getenv("API_KEY")
|
| 36 |
if not API_KEY:
|
| 37 |
-
raise ValueError("API_KEY
|
| 38 |
|
| 39 |
MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
|
| 40 |
ENV_BASE_URL = os.getenv("ENV_BASE_URL", "http://localhost:7860")
|
|
|
|
| 29 |
raise Exception(f"HTTP Error {e.code}: {e.read().decode('utf-8')}")
|
| 30 |
|
| 31 |
# ββ Environment variables ββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 32 |
+
# API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
|
| 33 |
+
# HF_TOKEN = os.getenv("HF_TOKEN")
|
| 34 |
+
|
| 35 |
+
# API_KEY = HF_TOKEN or os.getenv("API_KEY")
|
| 36 |
+
# if not API_KEY:
|
| 37 |
+
# raise ValueError("API_KEY environment variable is required")
|
| 38 |
+
|
| 39 |
+
API_BASE_URL = os.getenv("API_BASE_URL")
|
| 40 |
+
API_KEY = os.getenv("API_KEY")
|
| 41 |
+
|
| 42 |
+
if not API_BASE_URL:
|
| 43 |
+
raise ValueError("API_BASE_URL must be set")
|
| 44 |
|
|
|
|
| 45 |
if not API_KEY:
|
| 46 |
+
raise ValueError("API_KEY must be set")
|
| 47 |
|
| 48 |
MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
|
| 49 |
ENV_BASE_URL = os.getenv("ENV_BASE_URL", "http://localhost:7860")
|