debugging..

app.py

@@ -46,9 +46,9 @@ def get_file_size_in_mb(file_path):
 #     print(f"{filename} downloaded successfully.")
 
 
-print(get_file_size_in_mb("hubert.pt"))
-print(get_file_size_in_mb("acoustic.pt"))
-print(get_file_size_in_mb("hifigan.pt"))
+# print(get_file_size_in_mb("hubert.pt"))
+# print(get_file_size_in_mb("acoustic.pt"))
+# 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)
@@ -58,23 +58,23 @@ print(get_file_size_in_mb("hifigan.pt"))
 # hubert, acoustic, hifigan = None, None, None
 
 # Load the state dictionaries from the CPU-saved files
-hubert_state_dict = torch.load("hubert.pt")
-acoustic_state_dict = torch.load("acoustic.pt")
-hifigan_state_dict = torch.load("hifigan.pt")
-
-print(hubert_state_dict)
-print(acoustic_state_dict)
-print(hifigan_state_dict)
-exit
-# 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= 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')
 
 # Function to initialize models with CUDA
 # def initialize_models():