Update app.py

app.py

@@ -10,6 +10,7 @@ import gradio as gr
 import librosa
 import torch
 from fairseq import checkpoint_utils
+from fairseq.data import dictionary
 
 from config import Config
 from lib.infer_pack.models import (
@@ -100,19 +101,17 @@ def model_data(model_name):
 
 def load_hubert():
     global hubert_model
-    models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
-        ["hubert_base.pt"],
-        suffix="",
-    )
+    safe_globals = [dictionary.Dictionary]  # allow this class
+    with torch.serialization.safe_globals(safe_globals):
+        models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
+            ["hubert_base.pt"],
+            suffix="",
+        )
     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()
+    hubert_model = hubert_model.half() if config.is_half else hubert_model.float()
     return hubert_model.eval()
 
-
 print("Loading hubert model...")
 hubert_model = load_hubert()
 print("Hubert model loaded.")