Update main.py

main.py

@@ -198,7 +198,7 @@ def model_init():
     tokenizer = BertTokenizerFast(vocab_file='/home/user/app/./bert-base-zh/vocab.txt', sep_token="[SEP]", pad_token="[PAD]", cls_token="[CLS]")
     # # tokenizer = BertTokenizer(vocab_file=args.voca_path)
     model = Word_BERT()
-    # # model = model.load_state_dict(torch.load(args.model_path))
+    model.load_state_dict(torch.load('/home/user/app/./model.pth'))
     # model = model.to(device)
     model.eval()
     return tokenizer,model