technicolor commited on
Commit
a8fe220
·
1 Parent(s): 4301879

update demo

Browse files
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"]