DataBoySu commited on
Commit
bfed85f
·
1 Parent(s): 1a65601
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -76,7 +76,7 @@ def get_model_message(client: OpenAI, obs_dict: dict, history: List[str]) -> str
76
  return '{"action": {"action_type": "submit_decision", "decision": "CLEAR", "evidence_links": []}}'
77
 
78
  async def main() -> None:
79
- client = OpenAI(base_url=API_BASE_URL, api_key=API_KEY)
80
 
81
  # Initialize your environment natively for the baseline script
82
  env = AmlEnvironment()
 
76
  return '{"action": {"action_type": "submit_decision", "decision": "CLEAR", "evidence_links": []}}'
77
 
78
  async def main() -> None:
79
+ client = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
80
 
81
  # Initialize your environment natively for the baseline script
82
  env = AmlEnvironment()