UjjwalPardeshi commited on
Commit ·
9bb9fe6
1
Parent(s): ee366d9
fix: inference format
Browse files- 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")
|