Update app.py

app.py

@@ -7,7 +7,7 @@ import re
 
 def paragraph_leveling(text):
     model_name = "./trained_model/fine_tunning_encoder_v2"
-    model = AutoModel.from_pretrained(model_name).to('cpu')
+    model = AutoModel.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')
 
     class MLP(nn.Module):
@@ -23,7 +23,6 @@ def paragraph_leveling(text):
 
     classifier = MLP(target_size=3, input_size=768)
     classifier.load_state_dict(torch.load('./trained_model/fine_tunning_classifier', map_location=torch.device('cpu')))
-    classifier
     classifier.eval()
 
     output_list = []