Spaces:
Runtime error
Runtime error
Use one model first
Browse files
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'
|
|
|
|
| 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):
|