Spaces:
Runtime error
Runtime error
Fix: use LOCAL_IMAGE_NAME (matches submission form)
Browse files- inference.py +2 -2
inference.py
CHANGED
|
@@ -6,7 +6,7 @@ MANDATORY
|
|
| 6 |
API_BASE_URL The API endpoint for the LLM.
|
| 7 |
MODEL_NAME The model identifier to use for inference.
|
| 8 |
HF_TOKEN Your Hugging Face / API key.
|
| 9 |
-
IMAGE_NAME The name of the local image to use for the environment if using from_docker_image()
|
| 10 |
|
| 11 |
- The inference script must be named `inference.py` and placed in the root directory of the project
|
| 12 |
- Participants must use OpenAI Client for all LLM calls using above variables
|
|
@@ -50,7 +50,7 @@ from client import CodeReviewEnv
|
|
| 50 |
|
| 51 |
# βββ Configuration ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 52 |
|
| 53 |
-
IMAGE_NAME = os.getenv("IMAGE_NAME") # If using from_docker_image()
|
| 54 |
API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
|
| 55 |
|
| 56 |
API_BASE_URL = os.getenv("API_BASE_URL") or "https://router.huggingface.co/v1"
|
|
|
|
| 6 |
API_BASE_URL The API endpoint for the LLM.
|
| 7 |
MODEL_NAME The model identifier to use for inference.
|
| 8 |
HF_TOKEN Your Hugging Face / API key.
|
| 9 |
+
IMAGE_NAME The name of the local image to use for the environment if using from_docker_image() (also accepts LOCAL_IMAGE_NAME)
|
| 10 |
|
| 11 |
- The inference script must be named `inference.py` and placed in the root directory of the project
|
| 12 |
- Participants must use OpenAI Client for all LLM calls using above variables
|
|
|
|
| 50 |
|
| 51 |
# βββ Configuration ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 52 |
|
| 53 |
+
IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME") or os.getenv("IMAGE_NAME") # If using from_docker_image()
|
| 54 |
API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
|
| 55 |
|
| 56 |
API_BASE_URL = os.getenv("API_BASE_URL") or "https://router.huggingface.co/v1"
|