Update Model/MultimodelNER/app_MNER.py
Browse files
Model/MultimodelNER/app_MNER.py
CHANGED
|
@@ -22,7 +22,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 22 |
|
| 23 |
|
| 24 |
net = getattr(resnet, 'resnet152')()
|
| 25 |
-
net.load_state_dict(torch.load(os.path.join('
|
| 26 |
encoder = myResnet(net, True, device)
|
| 27 |
def process_text(text):
|
| 28 |
# Loại bỏ dấu cách thừa và dấu cách ở đầu và cuối văn bản
|
|
|
|
| 22 |
|
| 23 |
|
| 24 |
net = getattr(resnet, 'resnet152')()
|
| 25 |
+
net.load_state_dict(torch.load(os.path.join('Model/Resnet/', 'resnet152.pth')))
|
| 26 |
encoder = myResnet(net, True, device)
|
| 27 |
def process_text(text):
|
| 28 |
# Loại bỏ dấu cách thừa và dấu cách ở đầu và cuối văn bản
|