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):
|