ttj commited on
Commit
22cd91e
·
1 Parent(s): a6dd028

Use summarization

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ def get_pipe(name):
4
  tokenizer = AutoTokenizer.from_pretrained(name)
5
  model = AutoModelForSeq2SeqLM.from_pretrained(name)
6
  pipe = pipeline(
7
- "text-generation", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
  model_names = ['bigscience/T0_3B','bigscience/T0'] #, 'bigscience/T0p', 'bigscience/T0pp']
 
4
  tokenizer = AutoTokenizer.from_pretrained(name)
5
  model = AutoModelForSeq2SeqLM.from_pretrained(name)
6
  pipe = pipeline(
7
+ "summarization", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
  model_names = ['bigscience/T0_3B','bigscience/T0'] #, 'bigscience/T0p', 'bigscience/T0pp']