Update app.py

app.py

@@ -55,7 +55,6 @@ 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):
@@ -103,6 +102,7 @@ def main():
 		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()
 			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))