Update app.py

app.py

@@ -9,26 +9,16 @@ def setup():
         return
     
     print("Installing dependencies...")
-    os.system("pip install -q gfpgan realesrgan basicsr")
+    # Install compatible versions
+    os.system("pip install -q numpy==1.23.5")
+    os.system("pip install -q gfpgan realesrgan")
     
     print("Downloading model...")
-    os.system("git clone https://github.com/vinthony/video-retalking.git vrt")
+    os.system("git clone https://github.com/OpenTalker/video-retalking.git vrt")
     
     # Download checkpoints
     os.chdir("vrt")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/30_net_gen.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/BFM.zip -P checkpoints/")
-    os.system("unzip -q checkpoints/BFM.zip -d checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/DNet.pt -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/ENet.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/expression.mat -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/face3d_pretrain_epoch_20.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/GFPGANv1.3.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/GPEN-BFR-512.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/LNet.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/ParseNet-latest.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/RetinaFace-R50.pth -P checkpoints/")
-    os.system("wget https://github.com/vinthony/video-retalking/releases/download/v0.0.1/shape_predictor_68_face_landmarks.dat -P checkpoints/")
+    os.system("bash scripts/download_models.sh")
     os.chdir("..")
     
     Path("setup_done.txt").touch()
@@ -49,14 +39,14 @@ def generate(image, audio):
         result = os.system(cmd)
         
         if Path(output).exists():
-            return output, "✅ Done!"
+            return output, "✅ Video generated successfully!"
         else:
-            return None, "❌ Failed to generate"
+            return None, "❌ Failed to generate video"
             
     except Exception as e:
         return None, f"❌ Error: {str(e)}"
 
-# Simple Gradio UI
+# Simple UI
 demo = gr.Interface(
     fn=generate,
     inputs=[
@@ -68,7 +58,7 @@ demo = gr.Interface(
         gr.Textbox(label="Status")
     ],
     title="AI Lip Sync Generator",
-    description="Upload a face image and audio to generate lip-synced video"
+    description="Upload face image and audio"
 )
 
 demo.launch()