junhyun01 commited on
Commit
6bd5a8b
·
1 Parent(s): 4ebbbf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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).to('cpu')
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 = []