Spaces:

zliang
/

fastpaperlayout

Sleeping

zliang commited on May 26

Commit

91a1c99

•

1 Parent(s): 01666b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,5 @@
 import gradio as gr
 import numpy as np
 import fitz  # PyMuPDF

+import os
+os.system("sed -i 's/config.max_workspace_size = self.args.workspace \* 1 << 30/config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, self.args.workspace \* 1 << 30)/' exporter.py")
 import gradio as gr
 import numpy as np
 import fitz  # PyMuPDF