changed view

app.py

@@ -5,13 +5,12 @@ model_name = "gpt2-large"
 
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-@st.cache
-def load_pipeline(model_name):
-    
+@st.cache(allow_output_mutation=True)
+def load_model(model_name):
     model =  AutoModelForCausalLM.from_pretrained(model_name,pad_token_id=tokenizer.eos_token_id)
     return model
 
-model = load_pipeline(model_name)
+model = load_model(model_name)
 pipeline = pipeline('text-generation', model=model, tokenizer=tokenizer)
 print("loaded the model/pipeline")
 

requirements.txt

@@ -1,4 +1,4 @@
 transformers>=4.18.0
-streamlit
+streamlit==1.9.0
 torch
 sentencepiece