fix hubert loading

app.py

@@ -39,6 +39,7 @@ import math
 from utils import load_audio, CSVutil
 
 global DoFormant, Quefrency, Timbre
+from transformers import HubertModel, HubertConfig
 
 if not os.path.isdir('csvdb/'):
     os.makedirs('csvdb')
@@ -246,8 +247,11 @@ associated_links = {}
 def load_hubert():
     global hubert_model
     # Load the model
-    hubert_model = torch.load("hubert.pth", map_location=config.device)
-    
+
+    configH= HubertConfig()
+    configH.output_hidden_states = True
+    hubert_model = HubertModel(configH)
+    hubert_model.load_state_dict(torch.load('hubert.pth'))
     # Prepare the model
     hubert_model = hubert_model.to(config.device)
     if config.is_half: