diff --git a/docker-compose.yml b/docker-compose.yml index 4be6d4f..60ba5e3 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -39,6 +39,8 @@ services: - MAX_MODEL_LEN=2048 - CACHE_ENABLED=true - RETENTION_DAYS=10 + - HF_TOKEN=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 + - HUGGING_FACE_HUB_TOKEN=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 # Resource limits (adjust based on your Orin config) deploy: diff --git a/main.py b/main.py index a18d690..9c5c5e1 100644 --- a/main.py +++ b/main.py @@ -39,7 +39,7 @@ from fastapi.responses import FileResponse, StreamingResponse from pydantic import BaseModel # Configuration from environment -ORPHEUS_MODEL = os.getenv("ORPHEUS_MODEL", "medium-3b") +ORPHEUS_MODEL = os.getenv("ORPHEUS_MODEL", "canopylabs/orpheus-tts-0.1-finetune-prod") CACHE_ENABLED = os.getenv("CACHE_ENABLED", "true").lower() == "true" CACHE_DIR = Path(os.getenv("CACHE_DIR", "cache")) OUTPUT_DIR = Path(os.getenv("OUTPUT_DIR", "output"))