rishitha14 commited on
Commit
b8a7cb2
·
verified ·
1 Parent(s): 79b4876

Updated API

Browse files
Files changed (1) hide show
  1. inference.py +4 -4
inference.py CHANGED
@@ -29,10 +29,10 @@ load_dotenv()
29
  # Configuration
30
  # ---------------------------------------------------------------------------
31
 
32
- API_KEY = os.environ("API_KEY", "")
33
- API_BASE_URL = os.environ("API_BASE_URL", "https://router.huggingface.co/v1")
34
- MODEL_NAME = os.environ("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
35
- ENV_URL = os.environ("ENV_URL", "http://localhost:7860")
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