Lillian McAboy commited on
Commit
d5683ff
·
1 Parent(s): 5897d9e

added summarize

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ formality_name = st.selectbox("Transform my text to be: ", formal_options)
21
 
22
  @st.cache(allow_output_mutation=True)
23
  def get_tokenizer(model_name):
 
24
  return AutoTokenizer.from_pretrained(model_name)
25
 
26
  @st.cache(allow_output_mutation=True)
 
21
 
22
  @st.cache(allow_output_mutation=True)
23
  def get_tokenizer(model_name):
24
+ from transformers import AutoTokenizer
25
  return AutoTokenizer.from_pretrained(model_name)
26
 
27
  @st.cache(allow_output_mutation=True)