changed view

app.py

@@ -3,14 +3,16 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
   
 model_name = "gpt2-large"
 
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
 @st.cache
 def load_pipeline(model_name):
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    
     model =  AutoModelForCausalLM.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id)
-    return (tokenizer,model)
+    return model
 
-model_tokenizer = load_pipeline(model_name)
-pipeline = pipeline('text-generation', model=model_tokenizer[1], tokenizer=model_tokenizer[0])
+model = load_pipeline(model_name)
+pipeline = pipeline('text-generation', model=model, tokenizer=tokenizer)
 print("loaded the model/pipeline")
 
 default_value = "But not just any roof cleaning will do."