Update app.py

app.py

@@ -96,18 +96,22 @@ def model_data(model_name: str):
         raise
 
 def load_hubert():
-    """Memuat model Hubert."""
-    try:
-        from fairseq.fairseq import checkpoint_utils
-        models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
-            [str(BASE_DIR / "hubert_base.pt")], arg_overrides={"data": str(BASE_DIR)}
-        )
-        hubert_model = models[0].to(config.device)
-        hubert_model = hubert_model.half() if config.is_half else hubert_model.float()
-        return hubert_model.eval()
-    except Exception as e:
-        logging.error(f"Error memuat Hubert: {e}")
-        raise
+    from fairseq import fairseq
+    
+    forward_dml = fairseq.GradMultiply.forward 
+
+    
+    models, _, _ = fairseq.load_model(
+        f"{BASE_DIR}/hubert_base.pt",
+    )
+    
+    hubert_model = models[0]
+    hubert_model = hubert_model.to(config.device)
+    if config.is_half:
+        hubert_model = hubert_model.half()
+    else:
+        hubert_model = hubert_model.float()
+    return hubert_model.eval()
 
 def download_file(url: str, output_path: str = None):
     """Mengunduh file dari URL dengan progress bar."""