Update app.py

app.py

@@ -100,9 +100,9 @@ def main():
 	elif st.checkbox("Text Generation"):
 		st.subheader("Generate Text")
 		message = st.text_area("Enter the Text","Type please ..")
-		input_ids = tokenizer(message, return_tensors='pt').input_ids
 		if st.button("Generate"):
 		    tokenizer, model = load_models()
+		    input_ids = tokenizer(message, return_tensors='pt').input_ids
 		    st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
 		    output = model.generate(input_ids, max_length=128)
 		    st.success(tokenizer.decode(output[0], skip_special_tokens=True))