Update app.py
Browse files
app.py
CHANGED
@@ -14,6 +14,14 @@ import torch
|
|
14 |
from loguru import logger
|
15 |
import nltk
|
16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
# 設定 HTTPS context 避免證書錯誤
|
18 |
ssl._create_default_https_context = ssl._create_unverified_context
|
19 |
|
|
|
14 |
from loguru import logger
|
15 |
import nltk
|
16 |
|
17 |
+
import torch.serialization
|
18 |
+
torch.serialization.register_package("utils")
|
19 |
+
_original_torch_load = torch.load
|
20 |
+
def safe_torch_load(*args, **kwargs):
|
21 |
+
kwargs.setdefault("weights_only", False)
|
22 |
+
return _original_torch_load(*args, **kwargs)
|
23 |
+
torch.load = safe_torch_load
|
24 |
+
|
25 |
# 設定 HTTPS context 避免證書錯誤
|
26 |
ssl._create_default_https_context = ssl._create_unverified_context
|
27 |
|