diff --git a/docker-compose.yml b/docker-compose.yml index 338e355..0ebf9cb 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -3,35 +3,18 @@ services: restart: unless-stopped build: . environment: - # - RABBITMQ_PASS=${RABBITMQ_PASS} - - RABBITMQ_PASS=L@l321321321 - - RABBITMQ_HOST=154.12.229.181 - - RABBITMQ_PORT=32790 - # - GEMINI_API_KEY=${GEMINI_API_KEY} - - GEMINI_API_KEY=AIzaSyB5TPjSPPZG1Qb6EtblhKFAjvCOdY15rcw - - GEMINI_MODEL=${GEMINI_MODEL:-gemini-2.5-flash} - # - OPENROUTER_API_KEY=${OPENROUTER_API_KEY} - - OPENROUTER_API_KEY=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 - - OPENROUTER_MODEL=${OPENROUTER_MODEL:-openai/gpt-oss-20b:free} - - FASTER_WHISPER_MODEL_SIZE=${FASTER_WHISPER_MODEL_SIZE:-small} + - FASTER_WHISPER_MODEL_SIZE=medium + - GEMINI_MODEL=gemini-2.5-flash + - OPENROUTER_API_KEY=${OPENROUTER_API_KEY} + - OPENROUTER_MODEL=openai/gpt-oss-20b:free + - RABBITMQ_PASS=${RABBITMQ_PASS} volumes: - # - "/root/videos:/app/videos" - # - "/root/outputs:/app/outputs" - - "./videos:/app/videos" - - "./outputs:/app/outputs" + - "/root/videos:/app/videos" + - "/root/outputs:/app/outputs" command: "python -u main.py" - # runtime: nvidia + networks: + - dokploy-network - # networks: - # - dokploy-network - - # deploy: - # resources: - # reservations: - # devices: - # - driver: nvidia - # count: all - # capabilities: [gpu] -# networks: -# dokploy-network: -# external: true +networks: + dokploy-network: + external: true diff --git a/video_render/config.py b/video_render/config.py index 4940751..8f346ad 100644 --- a/video_render/config.py +++ b/video_render/config.py @@ -27,7 +27,7 @@ class RabbitMQSettings: @dataclass(frozen=True) class GeminiSettings: api_key: str = os.environ.get("GEMINI_API_KEY", "") - model: str = os.environ.get("GEMINI_MODEL", "gemini-2.5-pro") + model: str = os.environ.get("GEMINI_MODEL", "gemini-2.5-flash") safety_settings: str | None = os.environ.get("GEMINI_SAFETY_SETTINGS") temperature: float = float(os.environ.get("GEMINI_TEMPERATURE", 0.2)) top_k: int | None = (