Spaces:

zliang
/

fastpaperlayout

Sleeping

zliang commited on May 26

Commit

e573d70

•

1 Parent(s): 91a1c99

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
-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 os
+os.system("pip install -q git+https://github.com/THU-MIG/yolov10.git")
 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