debugging..

app.py

@@ -26,15 +26,20 @@ def get_file_size_in_mb(file_path):
 # 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 = hubert.HubertSoft
+
+hubert_loaded = torch.load("hubert_cpu.pt", map_location=torch.device('cpu'))
+# acoustic_loaded = torch.load("acoustic_cpu.pt", map_location=torch.device('cpu'))
+# hifigan_loaded = 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)
-# acoustic.load_state_dict(acoustic_state_dict.state_dict(),  strict=False)
-# hifigan.load_state_dict(hifigan_state_dict.state_dict(),  strict=False)
+hubert.load_state_dict(hubert_loaded.state_dict(),  strict=False)
+# acoustic.load_state_dict(acoustic_loaded.state_dict(),  strict=False)
+# hifigan.load_state_dict(hifigan_loaded.state_dict(),  strict=False)
+
 
+print(hubert_loaded)
+sys.exit()
 # Move models to CPU (if not already on CPU)
 # hubert = hubert.to('cpu')
 # acoustic = acoustic.to('cpu')