Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
 #Load model
-model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-en-pt-t5")
+model = AutoModelForSeq2SeqLM.from_pretrained('unicamp-dl/translation-en-pt-t5')
 
 #Tokenizer
-tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-en-pt-t5")
+tokenizer = AutoTokenizer.from_pretrained('unicamp-dl/translation-en-pt-t5')
 
 #Set up Translator
 translator = pipeline('text2text-generation', model=model, tokenizer=tokenizer)
@@ -21,6 +21,6 @@ txt_input = st.text_input('Enter text here:')
 if st.button('Submit'): 
     translation = translator(txt_input)
 
-#Display translation
-st.write(f'Translation: {translation}')
+    #Display translation
+    st.write(f'Translation: {translation}')