debugging..

app.py

@@ -18,14 +18,14 @@ def get_file_size_in_mb(file_path):
 
 
 # Load the models using torch.hub
-# hubert = torch.hub.load("bshall/hubert:main", "hubert_soft", trust_repo=True)
-# acoustic = torch.hub.load("bshall/acoustic-model:main", "hubert_soft", trust_repo=True)
-# hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True)
+hubert = torch.hub.load("bshall/hubert:main", "hubert_soft", trust_repo=True).to("cpu")
+acoustic = torch.hub.load("bshall/acoustic-model:main", "hubert_soft", trust_repo=True).to("cpu")
+hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True).to("cpu")
 
 # Load the state dictionaries from the CPU-saved files
-hubert = torch.load("hubert_cpu.pt", map_location=torch.device('cpu'))
-acoustic = torch.load("acoustic_cpu.pt", map_location=torch.device('cpu'))
-hifigan = torch.load("hifigan_cpu.pt", map_location=torch.device('cpu'))
+# hubert = torch.load("hubert_cpu.pt", map_location=torch.device('cpu'))
+# acoustic = torch.load("acoustic_cpu.pt", map_location=torch.device('cpu'))
+# hifigan = torch.load("hifigan_cpu.pt", map_location=torch.device('cpu'))
 
 # Set the state dictionaries to the models
 # hubert.load_state_dict(hubert_state_dict.state_dict(),  strict=False)