k

Dockerfile

@@ -27,9 +27,6 @@ RUN wget -P /app/SadTalker/checkpoints \
     https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2/epoch_20.pth \
     https://github.com/OpenTalker/SadTalker/releases/download/v0.0.2/facevid2vid_00189-model.pth.tar
 
-# Install SadTalker
-RUN pip install /app/SadTalker
-
 # Copy application files
 COPY . .
 
@@ -37,6 +34,9 @@ COPY . .
 RUN mkdir -p /app/static/uploads && chmod -R 777 /app/static/uploads
 RUN mkdir -p /app/templates && chmod -R 777 /app/templates
 
+# Add SadTalker to Python path
+ENV PYTHONPATH="${PYTHONPATH}:/app/SadTalker/src"
+
 # Force CPU mode
 ENV SADTALKER_FORCE_CPU=1
 EXPOSE 7860

app.py

@@ -1,15 +1,22 @@
 import os
+import sys
+sys.path.append('/app/SadTalker/src')
+
 from flask import Flask, render_template, request, jsonify
-from SadTalker import SadTalker
 
 app = Flask(__name__)
 
-# Initialize SadTalker with CPU
-sadtalker = SadTalker(
-    checkpoint_path="/app/SadTalker/checkpoints",
-    config_path="/app/SadTalker/src/config",
-    device="cpu"
-)
+# Initialize SadTalker with proper import
+try:
+    from inference import SadTalker
+    sadtalker = SadTalker(
+        checkpoint_path="/app/SadTalker/checkpoints",
+        config_path="/app/SadTalker/src/config",
+        device="cpu"
+    )
+except ImportError:
+    print("Warning: SadTalker not properly initialized")
+    sadtalker = None
 
 @app.route('/')
 def home():