Update app.py

app.py

@@ -37,7 +37,8 @@ def read_fn(path):
     disk_rw = DiskReaderWriter(os.path.dirname(path))
     return disk_rw.read(os.path.basename(path), AbsReaderWriter.MODE_BIN)
 
-@spaces.GPU(duration=120)
+
+@spaces.GPU
 def parse_pdf(doc_path, output_dir, end_page_id):
     os.makedirs(output_dir, exist_ok=True)
 
@@ -140,6 +141,27 @@ def show_pdf(file):
 latex_delimiters = [{"left": "$$", "right": "$$", "display": True},
                     {"left": '$', "right": '$', "display": False}]
 
+
+from magic_pdf.model.doc_analyze_by_custom_model import ModelSingleton
+
+def init_model():
+    from magic_pdf.model.doc_analyze_by_custom_model import ModelSingleton
+    try:
+        model_manager = ModelSingleton()
+        txt_model = model_manager.get_model(False, False)
+        logger.info(f"txt_model init final")
+        ocr_model = model_manager.get_model(True, False)
+        logger.info(f"ocr_model init final")
+        return 0
+    except Exception as e:
+        logger.exception(e)
+        return -1
+
+
+model_init = init_model()
+logger.info(f"model_init: {model_init}")
+
+
 if __name__ == "__main__":
     with gr.Blocks() as demo:
         with gr.Row():