soumi guria commited on
Commit
07bb165
Β·
unverified Β·
2 Parent(s): 3d4a1b5187302e

Merge pull request #1 from soumiguria/soumi

Browse files
Files changed (1) hide show
  1. 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 environment variable is required")
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")