Spaces:
Sleeping
Sleeping
Commit
·
a8fe220
1
Parent(s):
4301879
update demo
Browse files- scripts/additional_scripts.py +1 -1
- scripts/setup_env.py +1 -1
scripts/additional_scripts.py
CHANGED
@@ -7,7 +7,7 @@ import os
|
|
7 |
file_path = "/root/magic-pdf.json"
|
8 |
|
9 |
new_config = {
|
10 |
-
"device-mode": "cuda",
|
11 |
"layout-config": {
|
12 |
"model": "layoutlmv3"
|
13 |
},
|
|
|
7 |
file_path = "/root/magic-pdf.json"
|
8 |
|
9 |
new_config = {
|
10 |
+
# "device-mode": "cuda",
|
11 |
"layout-config": {
|
12 |
"model": "layoutlmv3"
|
13 |
},
|
scripts/setup_env.py
CHANGED
@@ -27,7 +27,7 @@ commands = [
|
|
27 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "markdown_pdf==1.3"],
|
28 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "bertopic==0.16.3"],
|
29 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "-U", "langchain-community"],
|
30 |
-
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "--force-reinstall", "torch==2.3.1", "torchvision==0.18.1", "numpy<2.0.0", "--index-url", "https://download.pytorch.org/whl/cu118"],
|
31 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "-U", "mineru[core]"],
|
32 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "Django==2.2.5"],
|
33 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "graphviz"]
|
|
|
27 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "markdown_pdf==1.3"],
|
28 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "bertopic==0.16.3"],
|
29 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "-U", "langchain-community"],
|
30 |
+
# [sys.executable, "-m", "pip", "install", "--no-cache-dir", "--force-reinstall", "torch==2.3.1", "torchvision==0.18.1", "numpy<2.0.0", "--index-url", "https://download.pytorch.org/whl/cu118"],
|
31 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "-U", "mineru[core]"],
|
32 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "Django==2.2.5"],
|
33 |
[sys.executable, "-m", "pip", "install", "--no-cache-dir", "graphviz"]
|