Pippoz commited on
Commit
65aee6e
1 Parent(s): 80882a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ st.markdown('## Text-generation OPT from Meta ')
7
 
8
  @st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
9
  def get_model():
10
- return pipeline('text-generation', model=model, skip_special_tokens=True)
11
 
12
  col1, col2 = st.columns([2,1])
13
 
 
7
 
8
  @st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
9
  def get_model():
10
+ return pipeline('text-generation', model=model, do_sample=True, skip_special_tokens=True)
11
 
12
  col1, col2 = st.columns([2,1])
13