Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 FROM python:3.10-slim
 
 # System dependencies
-RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y git wget && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /workspace
 
@@ -9,34 +9,89 @@ WORKDIR /workspace
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
-# Download models and adapters at build time
-# You can replace MODEL_ID or adapter repos as needed
+# Set Hugging Face cache for all model downloads (best practice for Spaces)
+ENV TRANSFORMERS_CACHE="/workspace/.cache/huggingface"
+RUN mkdir -p /workspace/.cache/huggingface
 
-ENV CACHE_DIR="/data"
-ENV HF_HOME="/data"
-# ENV TRANSFORMERS_CACHE="/data"
-# ENV DEEPFACE_HOME="/data"
-# ENV INSIGHTFACE_HOME="/data/.insightface"
-RUN mkdir -p /data && \
-    chmod -R 777 /data
+# Pre-download Stable Diffusion v1-5 weights into cache
+RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='/workspace/.cache/huggingface', torch_dtype=torch.float32)"
 
-# -- Download Stable Diffusion v1-5 weights --
-# RUN python -c "from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype='float32')"
-RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype=torch.float32)"
-
-
-RUN apt-get update && apt-get install -y wget
-
-# -- Download IP-Adapter weights (official Hugging Face repo) --
-RUN mkdir -p ./models/ip_adapter && \
-    wget -O ./models/ip_adapter/ip-adapter_sd15.bin https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15.bin
+# Pre-download IP-Adapter weights (if using)
+RUN mkdir -p /workspace/.cache/huggingface/ip_adapter && \
+    wget -O /workspace/.cache/huggingface/ip_adapter/ip-adapter_sd15.bin https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15.bin
 
 # Copy app code
 COPY . .
 
-# Set environment (disable gradio analytics, useful for Spaces)
+# Disable Gradio analytics (optional, but good for Spaces)
 ENV GRADIO_ANALYTICS_ENABLED="False"
 
 EXPOSE 7860
 
 CMD ["python", "app.py"]
+
+
+
+
+
+
+
+
+# FROM python:3.10-slim
+
+# # System dependencies
+# RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+
+# WORKDIR /workspace
+
+# # Install Python dependencies
+# COPY requirements.txt .
+# RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
+
+# # Download models and adapters at build time
+# # You can replace MODEL_ID or adapter repos as needed
+
+# ENV CACHE_DIR="/data"
+# ENV HF_HOME="/data"
+# # ENV TRANSFORMERS_CACHE="/data"
+# # ENV DEEPFACE_HOME="/data"
+# # ENV INSIGHTFACE_HOME="/data/.insightface"
+# RUN mkdir -p /data && \
+#     chmod -R 777 /data
+
+# # -- Download Stable Diffusion v1-5 weights --
+# # RUN python -c "from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype='float32')"
+# RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype=torch.float32)"
+
+
+# RUN apt-get update && apt-get install -y wget
+
+# # -- Download IP-Adapter weights (official Hugging Face repo) --
+# RUN mkdir -p ./models/ip_adapter && \
+#     wget -O ./models/ip_adapter/ip-adapter_sd15.bin https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15.bin
+
+
+# # Set Hugging Face cache for all model downloads (best practice for Spaces)
+# ENV TRANSFORMERS_CACHE="/workspace/.cache/huggingface"
+# RUN mkdir -p /workspace/.cache/huggingface && \
+#     chmod -R 777 /workspace
+
+# # Pre-download Stable Diffusion v1-5 weights into cache
+# RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='/workspace/.cache/huggingface', torch_dtype=torch.float32)"
+
+# # Pre-download IP-Adapter weights (if using)
+# RUN mkdir -p /workspace/.cache/huggingface/ip_adapter && \
+#     wget -O /workspace/.cache/huggingface/ip_adapter/ip-adapter_sd15.bin https://huggingface.co/h94/IP-Adapter/resolve/main/models/ip-adapter_sd15.bin
+
+
+
+
+# # Copy app code
+# COPY . .
+
+# # Set environment (disable gradio analytics, useful for Spaces)
+# ENV GRADIO_ANALYTICS_ENABLED="False"
+
+# EXPOSE 7860
+
+# CMD ["python", "app.py"]