YinuoGuo27 commited on
Commit
cd1584f
·
verified ·
1 Parent(s): 6e48b01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -12,6 +12,11 @@ local_dir = "./downloaded_repo"
12
  snapshot_download(repo_id=repo_id, local_dir=local_dir)
13
  print("Download complete! Repository saved at:", local_dir)
14
 
 
 
 
 
 
15
 
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
17
  torch.set_default_device(device)
@@ -44,7 +49,6 @@ def main(sadtaker_checkpoint_path=r"SadTalker/checkpoints", sadtalker_config_pat
44
  print('device is', device)
45
  torch.set_default_device(device)
46
  tts = TTS('tts_models/multilingual/multi-dataset/your_tts').to(device)
47
- sad_talker = SadTalker(sadtaker_checkpoint_path, sadtalker_config_path, lazy_load=True)
48
  kd_talker = Inferencer()
49
 
50
  with gr.Blocks(analytics_enabled=False) as interface:
 
12
  snapshot_download(repo_id=repo_id, local_dir=local_dir)
13
  print("Download complete! Repository saved at:", local_dir)
14
 
15
+ print("\nFiles downloaded:")
16
+ for root, dirs, files in os.walk(local_dir):
17
+ for file in files:
18
+ file_path = os.path.join(root, file)
19
+ print(file_path)
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  torch.set_default_device(device)
 
49
  print('device is', device)
50
  torch.set_default_device(device)
51
  tts = TTS('tts_models/multilingual/multi-dataset/your_tts').to(device)
 
52
  kd_talker = Inferencer()
53
 
54
  with gr.Blocks(analytics_enabled=False) as interface: