Update app.py

app.py

@@ -11,10 +11,6 @@ import torch
 #    return model, tokenizer
 generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)    
 
-@st.cache(allow_output_mutation=True)
-def define_model(prompt):  
-    answer = generator(prompt)
-    return answer
 
 
 #@st.cache(allow_output_mutation=True)
@@ -29,6 +25,7 @@ def define_model(prompt):
 
 prompt= st.text_area('Your prompt here',
  '''Hello, I'm am conscious and''')
+answer = generator(prompt, max_length=60)
  
 #answer = opt_model(prompt, model, tokenizer,)
 #lst = ['ciao come stai sjfsbd dfhsdf  fuahfuf  feuhfu wefwu ']