Update app.py

app.py

@@ -11,10 +11,9 @@ import torch
 #    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b", use_fast=False)
 #    return model, tokenizer
 
-@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None})
-def define_model():
+with st.spinner('Loading OPT-1.3b Model...'):
     generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)    
-    return generator
+st.success('Model loaded correctly!')
 
 #@st.cache(allow_output_mutation=True)
 #def opt_model(prompt, model, tokenizer, num_sequences = 1, max_length = 50):  
@@ -25,7 +24,6 @@ def define_model():
 
 
 #model, tokenizer = define_model()
-generator = define_model()
 
 prompt= st.text_area('Your prompt here',
  '''Hello, I'm am conscious and''')