Update app.py

app.py

@@ -14,26 +14,19 @@ import torch
 from loguru import logger
 import nltk
 
-# ✅ 修正後的 torch.load monkey patch，允許 utils.HParams 被載入
-import types
 import torch.serialization
+from torch.serialization import add_safe_globals
 
-_original_torch_load = torch.load
-
-# 模擬 parrots 模型裡用到的 utils.HParams 類別
 class DummyHParams:
     pass
 
-# 建立假的 utils 模組，讓 torch 找得到 utils.HParams
-dummy_utils = types.ModuleType("utils")
-dummy_utils.HParams = DummyHParams
-
-import sys
-sys.modules["utils"] = dummy_utils  # <-- 🔑 核心！告訴 Python：有一個叫 utils 的模組
+# ➕ 加入允許的類別進入 PyTorch 的安全清單中
+add_safe_globals([DummyHParams])
 
+# ✅ 繼續 monkey patch torch.load
+_original_torch_load = torch.load
 def patched_torch_load(*args, **kwargs):
-    with torch.serialization.safe_globals({"utils.HParams": DummyHParams}):
-        return _original_torch_load(*args, **kwargs)
+    return _original_torch_load(*args, **kwargs)
 
 torch.load = patched_torch_load