debugging..

app.py

@@ -18,26 +18,24 @@ 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, 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)
+# 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)
 
 # Load the state dictionaries from the CPU-saved files
-hubert_state_dict = torch.load("hubert_cpu.pt", map_location=torch.device('cpu'))
-acoustic_state_dict = torch.load("acoustic_cpu.pt", map_location=torch.device('cpu'))
-hifigan_state_dict = 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)
-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_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)
 
 # Move models to CPU (if not already on CPU)
-hubert = hubert.to('cpu')
-acoustic = acoustic.to('cpu')
-hifigan = hifigan.to('cpu')
+# hubert = hubert.to('cpu')
+# acoustic = acoustic.to('cpu')
+# hifigan = hifigan.to('cpu')
 
 
 
@@ -80,7 +78,6 @@ def enable_convert_button(audio):
         return gr.update(interactive=True)
     return gr.update(interactive=False)
 
-
 def clear_components():
   return None, None