Update Dockerfile

Dockerfile

@@ -15,10 +15,10 @@ RUN mkdir -p /home/user/app/hf_cache
 WORKDIR /home/user/app
 
 # کلون کردن ریپو CelebAMask-HQ
-RUN git clone https://github.com/switchablenorms/CelebAMask-HQ.git /home/user/app/CelebAMask-HQ
+RUN git clone https://github.com/switchablenorms/CelebAMask-HQ.git
 
 # تنظیم متغیرهای محیطی
-ENV PYTHONPATH="/home/user/app/CelebAMask-HQ:${PYTHONPATH}"
+ENV PYTHONPATH="/home/user/app/CelebAMask-HQ:/home/user/app/CelebAMask-HQ/face_parsing:${PYTHONPATH}"
 ENV HF_HOME="/home/user/app/hf_cache"
 ENV TRANSFORMERS_CACHE="/home/user/app/hf_cache"
 ENV HUGGINGFACE_HUB_CACHE="/home/user/app/hf_cache"
@@ -29,6 +29,10 @@ COPY --chown=user:user requirements.txt .
 # نصب پکیج‌ها
 RUN pip install --no-cache-dir -r requirements.txt
 
+# کپی فایل‌های اضافی مورد نیاز
+COPY --chown=user:user unet.py /home/user/app/CelebAMask-HQ/face_parsing/
+COPY --chown=user:user utils.py /home/user/app/CelebAMask-HQ/face_parsing/
+
 # کپی اپلیکیشن
 COPY --chown=user:user app.py .