Update app.py

app.py

@@ -96,10 +96,11 @@ def main():
 			st.success(TextBlob(message).correct())
 	if st.checkbox("Text Generation"):
 		st.subheader("Generate Text")
-		if st.button("Generate"):
-		    tokenizer, model = load_models()
+		ok = st.button("Generate")
+		tokenizer, model = load_models()
+		if ok:
 		    input_ids = tokenizer(message, return_tensors='pt').input_ids
-		    st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
+		    st.text("Using Hugging Face Transformer, Contrastive Search ..")
 		    output = model.generate(input_ids, max_length=128)
 		    st.success(tokenizer.decode(output[0], skip_special_tokens=True))
 	def change_photo_state():