DeepLearning101 commited on
Commit
6114804
·
verified ·
1 Parent(s): e4723ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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