Update app.py

app.py

@@ -1,3 +1,11 @@
+
+
+import torch
+from transformers import PegasusForConditionalGeneration, PegasusTokenizer
+
+import streamlit as st
+#from .paraphraser import get_paraphrased_sentences, model, tokenizer
+
 hide_main_and_footer_style = """
         <style>
         #MainMenu {visibility: hidden;} footer{visibility: hidden;}
@@ -7,10 +15,6 @@ hide_main_and_footer_style = """
 
 st.markdown(hide_main_and_footer_style, unsafe_allow_html=True)
 
-import torch
-from transformers import PegasusForConditionalGeneration, PegasusTokenizer
-
-
 model_name = 'tuner007/pegasus_paraphrase'
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
@@ -64,8 +68,6 @@ def summarize(text):
 #   return tokenizer.batch_decode(outputs, skip_special_tokens=True)
 
 
-import streamlit as st
-#from .paraphraser import get_paraphrased_sentences, model, tokenizer
 
 
 def app():