diff --git a/.env.sample b/.env.sample index b81eb6a..ee65daf 100644 --- a/.env.sample +++ b/.env.sample @@ -1,4 +1,4 @@ BROKER_BASE_URI=amqp://:@:5672 -DATABASE_URL= -STORAGE_BUCKET= -APP_NAME=text-to-art \ No newline at end of file +FIREBASE_DATABASE_URL= +FIREBASE_STORAGE_BUCKET= +FIREBASE_APP_NAME=text-to-art \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index afad815..38724f7 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -2,9 +2,7 @@ version: '3' services: stable-diffusion-v2: - build: - context: . - dockerfile: Dockerfile + image: devainetwork/tti-worker container_name: stable-diffuion-v2-worker env_file: - .env @@ -25,9 +23,7 @@ services: capabilities: [gpu] stable-diffusion-v2-1: - build: - context: . - dockerfile: Dockerfile + image: devainetwork/tti-worker container_name: stable-diffuion-v2-1-worker env_file: - .env diff --git a/src/configs/config.py b/src/configs/config.py index af35ff9..b43d734 100644 --- a/src/configs/config.py +++ b/src/configs/config.py @@ -15,8 +15,8 @@ class ModelSettings(BaseSettings): class FirebaseSettings(BaseSettings): cred_path: str = "/app/key/serviceAccountKey.json" - database_url: str - storage_bucket: str + firebase_database_url: str + firebase_storage_bucket: str app_name: str = "text-to-art" diff --git a/src/utils.py b/src/utils.py index 6ac93c0..e5f7dff 100644 --- a/src/utils.py +++ b/src/utils.py @@ -13,7 +13,7 @@ from schemas import ImageGenerationResponse, ImageGenerationResult, ImageGenerationWorkerOutput -app_name = firebase_settings.app_name +app_name = firebase_settings.firebase_app_name def clear_memory() -> None: diff --git a/src/worker.py b/src/worker.py index 5cf45f0..c6fa426 100644 --- a/src/worker.py +++ b/src/worker.py @@ -17,7 +17,7 @@ firebase_admin.initialize_app( cred, { - "databaseURL": firebase_settings.database_url, - "storageBucket": firebase_settings.storage_bucket, + "databaseURL": firebase_settings.firebase_database_url, + "storageBucket": firebase_settings.firebase_storage_bucket, }, )