ojasmov commited on
Commit
c0c9fed
·
verified ·
1 Parent(s): b8a7cb2

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -30,8 +30,8 @@ load_dotenv()
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
 
30
  # ---------------------------------------------------------------------------
31
 
32
  API_KEY = os.getenv("API_KEY", "")
33
+ API_BASE_URL = os.getenv("API_BASE_URL")
34
+ MODEL_NAME = os.getenv("MODEL_NAME")
35
  ENV_URL = os.getenv("ENV_URL", "http://localhost:7860")
36
  BENCHMARK = "xsecure"
37
  MAX_STEPS = 20