Update app.py

app.py

@@ -26,7 +26,6 @@ import edge_tts
 import gradio as gr
 import librosa
 import torch
-from fairseq import checkpoint_utils
 
 from config import Config
 from lib.infer_pack.models import (
@@ -125,10 +124,14 @@ def model_data(model_name):
 
 # Fungsi untuk memuat model Hubert
 def load_hubert():
-    models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
-        [f"{BASE_DIR}/hubert_base.pt"],
-        suffix="",
+    from fairseq import fairseq
+    fairseq.GradMultiply.forward = forward_dml
+    fairseq.modules.grad_multiply.GradMultiply.forward = forward_dml -> fairseq.GradMultiply.forward = forward_dml
+    
+    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: