Update app.py

app.py

@@ -23,7 +23,7 @@ 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.to('cpu')
+    classifier
     classifier.eval()
 
     output_list = []
@@ -45,7 +45,7 @@ def paragraph_leveling(text):
             padding='max_length',
             truncation=True,
             max_length=120)
-        output = model(**train_encoding.to('cpu'))
+        output = model(**train_encoding)
         output = classifier(output[1])
         output = output[0]
 
@@ -89,4 +89,4 @@ demo = gr.Interface(
     theme=gr.themes.Base()
 )
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()