Update app.py

app.py

@@ -2,6 +2,12 @@ import torch
 from transformers import T5ForConditionalGeneration,T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, PegasusTokenizer, PegasusForConditionalGeneration
 import streamlit as st
 
+st.title("Auto Translate (To English)")
+text = st.text_input("Okay")
+st.text("What you wrote: ")
+st.write(text)
+st.text("English Translation: ")
+
 model_name = 'tuner007/pegasus_paraphrase'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name)
@@ -14,11 +20,6 @@ def get_response(text):
   tgt_text = tokenizer.batch_decode(translated, skip_special_tokens=True)
   return tgt_text
     
-st.title("Auto Translate (To English)")
-text = st.text_input("Okay")
-st.text("What you wrote: ")
-st.write(text)
-st.text("English Translation: ")
 if text:
     translated_text = get_response(text)
     st.write(translated_text if translated_text else "No translation found")