Pratham commited on
Commit
113844e
Β·
1 Parent(s): dd270a1

update BASE_URL to live HF Space

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -8,7 +8,7 @@ from openai import OpenAI
8
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
9
  MODEL_NAME = os.getenv("MODEL_NAME", "meta-llama/Llama-3.1-8B-Instruct")
10
  HF_TOKEN = os.getenv("HF_TOKEN")
11
- BASE_URL = os.getenv("BASE_URL", "http://localhost:7860")
12
 
13
 
14
  # ── MANDATORY stdout format ───────────────────────────
 
8
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
9
  MODEL_NAME = os.getenv("MODEL_NAME", "meta-llama/Llama-3.1-8B-Instruct")
10
  HF_TOKEN = os.getenv("HF_TOKEN")
11
+ BASE_URL = os.getenv("BASE_URL", "https://endraode-pipeline-env.hf.space")
12
 
13
 
14
  # ── MANDATORY stdout format ───────────────────────────