Update app.py

app.py

@@ -14,22 +14,24 @@ import torch
 from loguru import logger
 import nltk
 
-# ✅ 加入 monkey patch 讓 torch.load 自動允許 utils.HParams
-import builtins
+# ✅ 修正後的 torch.load monkey patch，允許 utils.HParams 被載入
+import types
 import torch.serialization
+
 _original_torch_load = torch.load
 
-def patched_torch_load(*args, **kwargs):
-    import types
+# 模擬 parrots 模型裡用到的 utils.HParams 類別
+class DummyHParams:
+    pass
 
-    # 建立假的 utils 模組和 HParams 類別（只需要匹配名稱）
-    class DummyHParams:
-        pass
+# 建立假的 utils 模組，讓 torch 找得到 utils.HParams
+dummy_utils = types.ModuleType("utils")
+dummy_utils.HParams = DummyHParams
 
-    dummy_utils = types.ModuleType("utils")
-    dummy_utils.HParams = DummyHParams
+import sys
+sys.modules["utils"] = dummy_utils  # <-- 🔑 核心！告訴 Python：有一個叫 utils 的模組
 
-    # 用 safe_globals 臨時允許這個類別被 pickle 載入
+def patched_torch_load(*args, **kwargs):
     with torch.serialization.safe_globals({"utils.HParams": DummyHParams}):
         return _original_torch_load(*args, **kwargs)