Update app.py

app.py

@@ -14,13 +14,26 @@ import torch
 from loguru import logger
 import nltk
 
+# ✅ 加入 monkey patch 讓 torch.load 自動允許 utils.HParams
+import builtins
 import torch.serialization
-torch.serialization.register_package("utils")
 _original_torch_load = torch.load
-def safe_torch_load(*args, **kwargs):
-    kwargs.setdefault("weights_only", False)
-    return _original_torch_load(*args, **kwargs)
-torch.load = safe_torch_load
+
+def patched_torch_load(*args, **kwargs):
+    import types
+
+    # 建立假的 utils 模組和 HParams 類別（只需要匹配名稱）
+    class DummyHParams:
+        pass
+
+    dummy_utils = types.ModuleType("utils")
+    dummy_utils.HParams = DummyHParams
+
+    # 用 safe_globals 臨時允許這個類別被 pickle 載入
+    with torch.serialization.safe_globals({"utils.HParams": DummyHParams}):
+        return _original_torch_load(*args, **kwargs)
+
+torch.load = patched_torch_load
 
 # 設定 HTTPS context 避免證書錯誤
 ssl._create_default_https_context = ssl._create_unverified_context