pszemraj commited on
Commit
b40e029
โ€ข
1 Parent(s): 925dd67

๐Ÿ”Š de-pollute logs

Browse files

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

Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -178,10 +178,11 @@ if __name__ == "__main__":
178
  logging.info("Starting app instance")
179
 
180
  logging.info("Loading summ models")
181
- model, tokenizer = load_model_and_tokenizer("pszemraj/pegasus-x-large-book-summary")
182
- model_sm, tokenizer_sm = load_model_and_tokenizer(
183
- "pszemraj/long-t5-tglobal-base-16384-book-summary"
184
- )
 
185
 
186
  logging.info("Loading OCR model")
187
  with contextlib.redirect_stdout(None):
 
178
  logging.info("Starting app instance")
179
 
180
  logging.info("Loading summ models")
181
+ with contextlib.redirect_stdout(None):
182
+ model, tokenizer = load_model_and_tokenizer("pszemraj/pegasus-x-large-book-summary")
183
+ model_sm, tokenizer_sm = load_model_and_tokenizer(
184
+ "pszemraj/long-t5-tglobal-base-16384-book-summary"
185
+ )
186
 
187
  logging.info("Loading OCR model")
188
  with contextlib.redirect_stdout(None):