Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
π de-pollute logs
Browse filesSigned-off-by: peter szemraj <[email protected]>
app.py
CHANGED
|
@@ -178,10 +178,11 @@ if __name__ == "__main__":
|
|
| 178 |
logging.info("Starting app instance")
|
| 179 |
|
| 180 |
logging.info("Loading summ models")
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 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):
|