omm7 commited on
Commit
7c949ab
·
verified ·
1 Parent(s): b29b30b

Upload folder using huggingface_hub

Browse files
__pycache__/inference.cpython-314.pyc CHANGED
Binary files a/__pycache__/inference.cpython-314.pyc and b/__pycache__/inference.cpython-314.pyc differ
 
inference.py CHANGED
@@ -6,14 +6,15 @@ from typing import Any, Dict, List, Optional
6
  import requests
7
  from openai import OpenAI
8
 
9
- API_KEY = os.getenv("HF_TOKEN", "")
10
  API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
11
  MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o-mini")
 
12
  LOGENV_URL = os.getenv("LOGENV_URL", "http://localhost:7860")
13
  BENCHMARK = "NovaTechIncidentCommand"
14
  SUCCESS_THRESHOLD = 0.70
15
 
16
- client = OpenAI(api_key=API_KEY or "placeholder", base_url=API_BASE_URL)
17
 
18
 
19
  def log_start(task: str, env: str, model: str) -> None:
 
6
  import requests
7
  from openai import OpenAI
8
 
9
+ HF_TOKEN = os.getenv("HF_TOKEN")
10
  API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
11
  MODEL_NAME = os.getenv("MODEL_NAME", "gpt-4o-mini")
12
+ LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
13
  LOGENV_URL = os.getenv("LOGENV_URL", "http://localhost:7860")
14
  BENCHMARK = "NovaTechIncidentCommand"
15
  SUCCESS_THRESHOLD = 0.70
16
 
17
+ client = OpenAI(api_key=HF_TOKEN or "placeholder", base_url=API_BASE_URL)
18
 
19
 
20
  def log_start(task: str, env: str, model: str) -> None: