IZERE HIRWA Roger commited on
Commit
b94cede
·
1 Parent(s): 95c88d4
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +5 -2
app.py CHANGED
@@ -33,10 +33,10 @@ def generate():
33
  # Text-to-Speech (using gTTS)
34
  from gtts import gTTS
35
  tts = gTTS(text=text, lang='en')
36
-
37
  tts.save(audio_path)
38
-
39
- # Generate video (CPU optimized)
40
  sadtalker.generate(
41
  source_image=img_path,
42
  driven_audio=audio_path,
 
33
  # Text-to-Speech (using gTTS)
34
  from gtts import gTTS
35
  tts = gTTS(text=text, lang='en')
36
+
37
  tts.save(audio_path)
38
+
39
+ # Generate video (CPU optimized )
40
  sadtalker.generate(
41
  source_image=img_path,
42
  driven_audio=audio_path,
requirements.txt CHANGED
@@ -3,8 +3,11 @@ Flask==2.3.2
3
  gTTS==2.3.2
4
  ffmpeg-python==0.2.0
5
  numpy==1.24.3
6
- torch==2.0.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
7
- torchvision==0.15.2+cpu --extra-index-url https://download.pytorch.org/whl/cpu
 
 
 
8
 
9
  # SadTalker dependencies
10
  face-alignment==1.3.5
 
3
  gTTS==2.3.2
4
  ffmpeg-python==0.2.0
5
  numpy==1.24.3
6
+
7
+ # PyTorch CPU-only versions
8
+ --extra-index-url https://download.pytorch.org/whl/cpu
9
+ torch==2.0.1
10
+ torchvision==0.15.2
11
 
12
  # SadTalker dependencies
13
  face-alignment==1.3.5