Update app.py

app.py

@@ -10,7 +10,7 @@ import torch
 #    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b", use_fast=False)
 #    return model, tokenizer
 
-@st.cache(suppress_st_warning=True)
+@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None})
 def define_model():
     generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)    
     return generator
@@ -28,6 +28,7 @@ generator = define_model()
 
 prompt= st.text_area('Your prompt here',
  '''Hello, I'm am conscious and''') 
+ 
 answer = generator(prompt, max_length=100,no_repeat_ngram_size=3, early_stopping=True, num_beams=10)
  
 #answer = opt_model(prompt, model, tokenizer,)