Update app.py

app.py

@@ -8,7 +8,6 @@ import re
 def paragraph_leveling(text):
     model_name = "./trained_model/fine_tunning_encoder_v2"
     model = AutoModel.from_pretrained(model_name)
-    model.to('cuda')
     tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')
 
     class MLP(nn.Module):
@@ -24,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'))
-    classifier.cuda()
     classifier.eval()
 
     output_list = []
@@ -46,7 +44,7 @@ def paragraph_leveling(text):
             padding='max_length',
             truncation=True,
             max_length=120)
-        output = model(**train_encoding.to('cuda'))
+        output = model(**train_encoding)
         output = classifier(output[1])
         output = output[0]