chore: Update Dockerfile and requirements.txt, and add config_settings.py

Former-commit-id: 002a6825814239d28e518dd3d2f207c9dad8f5ae
Former-commit-id: 9a25f9401c97ac0ca7f91db319cd2cbee7ac528c

Dockerfile

@@ -1,7 +1,4 @@
-# Dockerfile for Python application with dependencies and runtime optimizations
-
-# Use Python 3.10 slim-buster as base image
-FROM python:3.10-slim-buster
+FROM python:3.10
 
 # Set working directory in the container
 WORKDIR /app

config_settings.py

@@ -0,0 +1,12 @@
+from pydantic import BaseSettings
+
+class Settings(BaseSettings):
+    AWS_ACCESS_KEY_ID: str = ''
+    AWS_SECRET_ACCESS_KEY: str = ''
+    AWS_REGION: str = "ap-south-1"
+    AWS_BUCKET_NAME: str="diffusion-model-bucket"
+
+   
+   
+Settings()
+

scripts/config.py

@@ -1,15 +1,17 @@
-MODEL_NAME="stabilityai/stable-diffusion-xl-base-1.0"
-ADAPTER_NAME = "VikramSingh178/sdxl-lora-finetune-product-caption"
-ADAPTER_NAME_2 = "VikramSingh178/Products10k-SDXL-Lora"
-VAE_NAME= "madebyollin/sdxl-vae-fp16-fix"
-DATASET_NAME= "hahminlew/kream-product-blip-captions"
-PROJECT_NAME = "Product Photography"
-PRODUCTS_10k_DATASET = "VikramSingh178/Products-10k-BLIP-captions"
-CAPTIONING_MODEL_NAME = "Salesforce/blip-image-captioning-base"
-SEGMENTATION_MODEL_NAME = "facebook/sam-vit-large"
-DETECTION_MODEL_NAME = "yolov8l"
-ENABLE_COMPILE = True
-INPAINTING_MODEL_NAME = 'kandinsky-community/kandinsky-2-2-decoder-inpaint'
+MODEL_NAME:str="stabilityai/stable-diffusion-xl-base-1.0"
+ADAPTER_NAME:str = "VikramSingh178/sdxl-lora-finetune-product-caption"
+ADAPTER_NAME_2:str = "VikramSingh178/Products10k-SDXL-Lora"
+VAE_NAME:str= "madebyollin/sdxl-vae-fp16-fix"
+DATASET_NAME:str = "hahminlew/kream-product-blip-captions"
+PROJECT_NAME:str = "Product Photography"
+PRODUCTS_10k_DATASET:str = "VikramSingh178/Products-10k-BLIP-captions"
+CAPTIONING_MODEL_NAME:str = "Salesforce/blip-image-captioning-base"
+SEGMENTATION_MODEL_NAME:str = "facebook/sam-vit-large"
+DETECTION_MODEL_NAME:str = "yolov8l"
+ENABLE_COMPILE:bool = True
+INPAINTING_MODEL_NAME:str = 'kandinsky-community/kandinsky-2-2-decoder-inpaint'
+
+
 
 
 
@@ -68,3 +70,5 @@ class Config:
         self.noise_offset = 0
         self.rank = 4
         self.debug_loss = False
+
+