debugging..

app.py

@@ -51,30 +51,26 @@ def get_file_size_in_mb(file_path):
 # print(get_file_size_in_mb("hifigan.pt"))
 
 # Load the models using torch.hub
-# hubert = torch.hub.load("bshall/hubert:main", "hubert_soft", trust_repo=True, pretrained=False)
-# acoustic = torch.hub.load("bshall/acoustic-model:main", "hubert_soft", trust_repo=True,  pretrained=False)
-# hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True, pretrained=False)
+hubert = torch.hub.load("bshall/hubert:main", "hubert_soft", trust_repo=True, pretrained=False)
+acoustic = torch.hub.load("bshall/acoustic-model:main", "hubert_soft", trust_repo=True,  pretrained=False)
+hifigan = torch.hub.load("bshall/hifigan:main", "hifigan_hubert_soft", trust_repo=True, pretrained=False)
 
 # hubert, acoustic, hifigan = None, None, None
 
 # Load the state dictionaries from the CPU-saved files
-hubert= torch.load("hubert.pt")
-acoustic= torch.load("acoustic.pt")
-hifigan= torch.load("hifigan.pt")
-
-# print(hubert_state_dict)
-# print(acoustic_state_dict)
-# print(hifigan_state_dict)
-# # Set the state dictionaries to the models
-
-# hubert.load_state_dict(hubert_state_dict)
-# acoustic.load_state_dict(acoustic_state_dict)
-# hifigan.load_state_dict(hifigan_state_dict)
-
-# # Move models to CPU (if not already on CPU)
-# hubert = hubert.to('cpu')
-# acoustic = acoustic.to('cpu')
-# hifigan = hifigan.to('cpu')
+hubert_state_dict = torch.load("hubert.pt", map_location=torch.device('cpu'))
+acoustic_state_dict = torch.load("acoustic.pt", map_location=torch.device('cpu'))
+hifigan_state_dict = torch.load("hifigan.pt", map_location=torch.device('cpu'))
+
+# Set the state dictionaries to the models
+hubert.load_state_dict(hubert_state_dict)
+acoustic.load_state_dict(acoustic_state_dict)
+hifigan.load_state_dict(hifigan_state_dict)
+
+# Move models to CPU (if not already on CPU)
+hubert = hubert.to('cpu')
+acoustic = acoustic.to('cpu')
+hifigan = hifigan.to('cpu')
 
 # Function to initialize models with CUDA
 # def initialize_models():