aryxn323 commited on
Commit
32167ee
·
verified ·
1 Parent(s): b35f1ab

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +4 -4
inference.py CHANGED
@@ -28,10 +28,10 @@ except ImportError:
28
  # Configuration
29
  # ---------------------------------------------------------------------------
30
 
31
- BASE_URL = os.environ.get("FRONTIER_ENV_URL", "http://localhost:7860")
32
- API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
33
- HF_TOKEN = os.environ.get("HF_TOKEN", "")
34
- MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o")
35
  MAX_STEPS_PER_TASK = 15
36
 
37
  # ---------------------------------------------------------------------------
 
28
  # Configuration
29
  # ---------------------------------------------------------------------------
30
 
31
+ BASE_URL = os.getenv("FRONTIER_ENV_URL", "http://localhost:7860")
32
+ API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
33
+ MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o")
34
+ HF_TOKEN = os.getenv("HF_TOKEN")
35
  MAX_STEPS_PER_TASK = 15
36
 
37
  # ---------------------------------------------------------------------------