keithhon commited on
Commit
9b25ecd
1 Parent(s): c1a5407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def get_pipe(name):
7
  "summarization", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
- model_names = ['bigscience/T0_3B'] #, 'bigscience/T0p', 'bigscience/T0pp']
11
  #model_names = ['bigscience/T0_3B','bigscience/T0'] #, 'bigscience/T0p', 'bigscience/T0pp']
12
  pipes = [get_pipe(name) for name in model_names]
13
  def _fn(text, do_sample, min_length, max_length, temperature, top_p, pipe):
 
7
  "summarization", model=model, tokenizer=tokenizer, framework="pt"
8
  )
9
  return pipe
10
+ model_names = ['bigscience/T0pp'] #, 'bigscience/T0p', 'bigscience/T0pp']
11
  #model_names = ['bigscience/T0_3B','bigscience/T0'] #, 'bigscience/T0p', 'bigscience/T0pp']
12
  pipes = [get_pipe(name) for name in model_names]
13
  def _fn(text, do_sample, min_length, max_length, temperature, top_p, pipe):