twigs commited on
Commit
b48a5e1
1 Parent(s): bfb18f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,9 +19,9 @@ simpl_tok = BartTokenizer.from_pretrained('twigs/bart-text2text-simplifier')
19
  simpl_model = BartForConditionalGeneration.from_pretrained(
20
  'twigs/bart-text2text-simplifier')
21
  cwi_pipe = pipeline('text-classification', model=cwi_model,
22
- tokenizer=cwi_tok, function_to_apply='none', device=0)
23
  fill_pipe = pipeline('fill-mask', model=simpl_model,
24
- tokenizer=simpl_tok, top_k=1, device=0)
25
 
26
 
27
  def id_replace_complex(s, threshold=0.4):
 
19
  simpl_model = BartForConditionalGeneration.from_pretrained(
20
  'twigs/bart-text2text-simplifier')
21
  cwi_pipe = pipeline('text-classification', model=cwi_model,
22
+ tokenizer=cwi_tok, function_to_apply='none')
23
  fill_pipe = pipeline('fill-mask', model=simpl_model,
24
+ tokenizer=simpl_tok, top_k=1)
25
 
26
 
27
  def id_replace_complex(s, threshold=0.4):