Update app.py

app.py

@@ -13,7 +13,7 @@ import torch
 @st.cache(suppress_st_warning=True)
 def define_model():
     generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)    
-
+    return generator
 
 #@st.cache(allow_output_mutation=True)
 #def opt_model(prompt, model, tokenizer, num_sequences = 1, max_length = 50):  
@@ -24,6 +24,7 @@ def define_model():
 
 
 #model, tokenizer = define_model()
+generator = define_model()
 
 prompt= st.text_area('Your prompt here',
  '''Hello, I'm am conscious and''')