Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import re
|
|
7 |
|
8 |
def paragraph_leveling(text):
|
9 |
model_name = "./trained_model/fine_tunning_encoder_v2"
|
10 |
-
model = AutoModel.from_pretrained(model_name)
|
11 |
tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')
|
12 |
|
13 |
class MLP(nn.Module):
|
@@ -23,7 +23,6 @@ def paragraph_leveling(text):
|
|
23 |
|
24 |
classifier = MLP(target_size=3, input_size=768)
|
25 |
classifier.load_state_dict(torch.load('./trained_model/fine_tunning_classifier', map_location=torch.device('cpu')))
|
26 |
-
classifier
|
27 |
classifier.eval()
|
28 |
|
29 |
output_list = []
|
|
|
7 |
|
8 |
def paragraph_leveling(text):
|
9 |
model_name = "./trained_model/fine_tunning_encoder_v2"
|
10 |
+
model = AutoModel.from_pretrained(model_name)
|
11 |
tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')
|
12 |
|
13 |
class MLP(nn.Module):
|
|
|
23 |
|
24 |
classifier = MLP(target_size=3, input_size=768)
|
25 |
classifier.load_state_dict(torch.load('./trained_model/fine_tunning_classifier', map_location=torch.device('cpu')))
|
|
|
26 |
classifier.eval()
|
27 |
|
28 |
output_list = []
|