Fix model download URLs and update requirements

app.py

@@ -35,9 +35,22 @@ FACE_SEG_MODEL = "face_segmentation.pth"
 # Direct download links for model weights.  These links point to publicly
 # accessible mirrors.  If they stop working, you can upload the files into
 # the checkpoints folder manually and the app will skip downloading.
+# Public mirrors for the required model weights.
+#
+# The original Wav2Lip checkpoint (``wav2lip_gan.pth``) and face segmentation
+# model are large binary files hosted on Hugging Face.  During development we
+# discovered that the URLs previously used in this script pointed at the wrong
+# repository (``Wave2lip`` vs ``Wave2Lip``) and returned 404 errors.  The
+# corrected links below refer to the official ``Wave2lip`` repository for the
+# GAN checkpoint and to a Wav2Lip‑HD repository for the face segmentation
+# model.  Note that these files are large (hundreds of megabytes) and may
+# require you to upload them manually into the ``checkpoints`` directory of
+# your Space if the automated download fails due to network restrictions.
 MODEL_URLS = {
+    # 436 MB GAN checkpoint hosted by Non‑playing‑Character
     WAV2LIP_MODEL: "https://huggingface.co/Non-playing-Character/Wave2lip/resolve/main/wav2lip_gan.pth",
-    FACE_SEG_MODEL: "https://huggingface.co/Non-playing-Character/Wave2lip/resolve/main/face_segmentation.pth",
+    # 53 MB face segmentation model hosted by commanderx (Wav2Lip‑HD project)
+    FACE_SEG_MODEL: "https://huggingface.co/commanderx/Wav2Lip-HD/resolve/main/checkpoints/face_segmentation.pth",
 }
 
 

requirements.txt

@@ -1,15 +1,25 @@
+# Pin gradio version for consistent API behaviour
 gradio==4.43.0
+
+# Core numerical libraries
 numpy
-opencv-python
-scipy
+
+# Audio processing
 pydub
+
+# Video and audio I/O helpers
 moviepy
+
+# Deep learning frameworks required by Wav2Lip
 torch
 torchvision
-gitpython
-ffmpeg-python
-tqdm
+
+# Utility libraries
 requests
-gfpgan
-realesrgan
-basicsr
+tqdm
+
+# FFmpeg bindings (used by moviepy/pydub).  Note: the FFmpeg binary is provided by the Spaces environment.
+ffmpeg-python
+
+# Optional: OpenCV for future enhancements (not strictly required by the current app but lightweight)
+opencv-python