ttj commited on
Commit
84b2150
1 Parent(s): b9010b8

Use one model first

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