Pippoz commited on
Commit
18f5ef8
·
1 Parent(s): da23edb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -13,7 +13,6 @@ import torch
13
  @st.cache(suppress_st_warning=True)
14
  def define_model():
15
  generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)
16
- return generator
17
 
18
 
19
  #@st.cache(allow_output_mutation=True)
 
13
  @st.cache(suppress_st_warning=True)
14
  def define_model():
15
  generator = pipeline('text-generation', model="facebook/opt-1.3b", skip_special_tokens=True)
 
16
 
17
 
18
  #@st.cache(allow_output_mutation=True)