UjjwalPardeshi commited on
Commit
9bb9fe6
·
1 Parent(s): ee366d9

fix: inference format

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -39,7 +39,7 @@ from openenv.core import GenericAction, GenericEnvClient
39
  # ---------------------------------------------------------------------------
40
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
41
  MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o")
42
- API_KEY = os.environ.get("HF_TOKEN") or os.environ.get("OPENAI_API_KEY", "")
43
  ENV_URL = os.environ.get("ENV_URL", "http://localhost:7860")
44
  IMAGE_NAME = os.environ.get("IMAGE_NAME", "")
45
  TASK_NAME = os.environ.get("TASK_NAME", "task_001")
 
39
  # ---------------------------------------------------------------------------
40
  API_BASE_URL = os.environ.get("API_BASE_URL", "https://api.openai.com/v1")
41
  MODEL_NAME = os.environ.get("MODEL_NAME", "gpt-4o")
42
+ API_KEY = os.environ.get("API_KEY") or os.environ.get("HF_TOKEN") or os.environ.get("OPENAI_API_KEY", "")
43
  ENV_URL = os.environ.get("ENV_URL", "http://localhost:7860")
44
  IMAGE_NAME = os.environ.get("IMAGE_NAME", "")
45
  TASK_NAME = os.environ.get("TASK_NAME", "task_001")