v9

Dockerfile

@@ -24,11 +24,15 @@ RUN useradd -m -u 1000 user && \
 USER user
 WORKDIR /home/user/app
 
+
 # Copy files and install only Gradio
 COPY --chown=user:user . /home/user/app
-RUN pip install --user pydantic==2.8.2 gradio 
 
-RUN pip install --user torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 
+RUN pip install --user --no-cache-dir -r requirements.txt
+
+COPY download_models.sh .
+RUN chmod +x download_models.sh
+RUN ./download_models.sh
 
 EXPOSE 7860
 CMD ["python", "app.py"]

app.py

@@ -3,22 +3,24 @@ import gradio as gr
 import argparse
 import cv2
 import time
+import os
+from utils.inference.image_processing import crop_face, get_final_image
+from utils.inference.video_processing import read_video, get_target, get_final_video, add_audio_from_another_video, face_enhancement
+from utils.inference.core import model_inference
+from network.AEI_Net import AEI_Net
+from coordinate_reg.image_infer import Handler
+from insightface_func.face_detect_crop_multi import Face_detect_crop
+from arcface_model.iresnet import iresnet100 # type: ignore[import-untyped]
+from models.pix2pix_model import Pix2PixModel
+from models.config_sr import TestOptions
+
 def setup_dependancies():
     if not os.path.exists("./download_models.sh"):
         print("Error: download_models.sh script not found.")
         sys.exit(1)
 
     subprocess.run(["./download_models.sh"])
-    
-    pkg_dir = os.path.expanduser("~/.local/gpu_packages")
-    os.makedirs(pkg_dir, exist_ok=True)
-    # subprocess.run([
-    #     sys.executable, "-m", "pip", "install",
-    #     "--upgrade", "--target", pkg_dir,
-    #     "torch", "torchvision", "torchaudio",
-    #     "--extra-index-url", "https://download.pytorch.org/whl/cu118", "mxnet-cu112", "onnxruntime-gpu==1.12", "Cython", "insightface==0.2.1", "kornia==0.5.4", "dill", "numpy"
-    # ], check=True)
-    sys.path.insert(0, pkg_dir)
+
 
 def init_models(args):
     

requirements.txt

@@ -1,10 +1,16 @@
-opencv-python
+torch 
+torchvision 
+torchaudio
+--index-url https://download.pytorch.org/whl/cu118
+pydantic==2.8.2
+gradio
+mxnet-cu112 
 onnx
-onnxruntime-gpu==1.12
-mxnet-cu112
+onnxruntime-gpu==1.12 
+Cython 
+insightface==0.2.1 
+kornia==0.5.4 dill
+opencv-python
 scikit-image
-insightface==0.2.1
 requests==2.25.1
-kornia==0.5.4
-dill
 wandb