pszemraj commited on
Commit
b73b513
1 Parent(s): a545c7f

♻️ generalize size

Browse files

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -82,8 +82,8 @@ def proc_submission(
82
 
83
  _summaries = summarize_via_tokenbatches(
84
  tr_in,
85
- model_sm if model_size == "base" else model,
86
- tokenizer_sm if model_size == "base" else tokenizer,
87
  batch_length=token_batch_length,
88
  **settings,
89
  )
 
82
 
83
  _summaries = summarize_via_tokenbatches(
84
  tr_in,
85
+ model_sm if "base" in model_size.lower() else model,
86
+ tokenizer_sm if "base" in model_size.lower() else tokenizer,
87
  batch_length=token_batch_length,
88
  **settings,
89
  )