Update app.py
Browse files
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/
|
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):
|