Update app.py

app.py

@@ -55,6 +55,7 @@ def load_models():
     tokenizer = AutoTokenizer.from_pretrained('gpt2-large')
     model = GPT2LMHeadModel.from_pretrained('gpt2-large')
     return tokenizer, model
+tokenizer, model = load_models()
 # Function For Extracting Entities
 @st.experimental_singleton
 def entity_analyzer(my_text):
@@ -100,7 +101,6 @@ def main():
 	elif st.checkbox("Text Generation"):
 		st.subheader("Generate Text")
 		message = st.text_area("Enter the Text","Type please ..")
-		tokenizer, model = load_models()
 		input_ids = tokenizer(message, return_tensors='pt').input_ids
 		if st.button("Generate"):
 			st.text("Using Hugging Face Trnsformer, Contrastive Search ..")