Update app.py

app.py

@@ -6,7 +6,7 @@ import re
 
 
 def paragraph_leveling(text):
-    model_name = "./trained_model/contrastive_encoder_sentence"
+    model_name = "contrastive_encoder_sentence"
     model = AutoModel.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained('zzxslp/RadBERT-RoBERTa-4m')
 
@@ -22,7 +22,7 @@ def paragraph_leveling(text):
             return out
 
     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.load_state_dict(torch.load('fine_tunning_classifier', map_location=torch.device('cpu')))
     classifier.eval()
 
     output_list = []