diff --git a/models/amazon-chronos/Dockerfile b/models/amazon-chronos/Dockerfile index fca1f4d..128113b 100644 --- a/models/amazon-chronos/Dockerfile +++ b/models/amazon-chronos/Dockerfile @@ -8,8 +8,8 @@ RUN pip install --no-cache-dir -r /usr/local/app/requirements.txt COPY ./models /usr/local/app/models COPY ./src /usr/local/app/src -ENV MODEL_NAME="chronos-t5-tiny" -ENV NUM_SAMPLES="20" +ENV ENFOBENCH_MODEL_NAME="chronos-t5-tiny" +ENV ENFOBENCH_NUM_SAMPLES="20" EXPOSE 3000 CMD ["uvicorn", "src.main:app", "--host", "0.0.0.0", "--port", "3000"] diff --git a/models/amazon-chronos/src/__init__.py b/models/amazon-chronos/src/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/models/amazon-chronos/src/main.py b/models/amazon-chronos/src/main.py index bb4126e..5c259cd 100644 --- a/models/amazon-chronos/src/main.py +++ b/models/amazon-chronos/src/main.py @@ -76,8 +76,8 @@ def forecast( return forecast -model_name = os.getenv("MODEL_NAME") -num_samples = int(os.getenv("NUM_SAMPLES")) +model_name = os.getenv("ENFOBENCH_MODEL_NAME") +num_samples = int(os.getenv("ENFOBENCH_NUM_SAMPLES")) # Instantiate your model model = AmazonChronosModel(model_name=model_name, num_samples=num_samples)