Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from pathlib import Path
|
|
8 |
import re
|
9 |
|
10 |
os.system('pip install -r requirements.txt')
|
11 |
-
os.system('pip install -U magic_pdf-0.8.
|
12 |
os.system('python -m pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/')
|
13 |
|
14 |
from huggingface_hub import snapshot_download
|
@@ -19,9 +19,7 @@ os.system('cp magic-pdf.template.json ~/magic-pdf.json')
|
|
19 |
os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /home/user/magic-pdf.json")
|
20 |
os.system("sed -i 's|cpu|cuda|g' /home/user/magic-pdf.json")
|
21 |
|
22 |
-
os.system('cp
|
23 |
-
os.system('cp ch_PP-OCRv4_rec_infer.tar /home/user/.paddleocr/whl/rec/ch/ch_PP-OCRv4_rec_infer/ch_PP-OCRv4_rec_infer.tar')
|
24 |
-
os.system('cp ch_ppocr_mobile_v2.0_cls_infer.tar /home/user/.paddleocr/whl/cls/ch_ppocr_mobile_v2.0_cls_infer/ch_ppocr_mobile_v2.0_cls_infer.tar')
|
25 |
|
26 |
os.system("pip install gradio-pdf")
|
27 |
from gradio_pdf import PDF
|
@@ -154,16 +152,16 @@ def init_model():
|
|
154 |
model_manager = ModelSingleton()
|
155 |
txt_model = model_manager.get_model(False, False)
|
156 |
logger.info(f"txt_model init final")
|
157 |
-
ocr_model = model_manager.get_model(True, False)
|
158 |
-
logger.info(f"ocr_model init final")
|
159 |
return 0
|
160 |
except Exception as e:
|
161 |
logger.exception(e)
|
162 |
return -1
|
163 |
|
164 |
|
165 |
-
|
166 |
-
|
167 |
|
168 |
|
169 |
if __name__ == "__main__":
|
|
|
8 |
import re
|
9 |
|
10 |
os.system('pip install -r requirements.txt')
|
11 |
+
os.system('pip install -U magic_pdf-0.8.0a3-py3-none-any.whl')
|
12 |
os.system('python -m pip install paddlepaddle-gpu==3.0.0b1 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/')
|
13 |
|
14 |
from huggingface_hub import snapshot_download
|
|
|
19 |
os.system(f"sed -i 's|/tmp/models|{model_dir}/models|g' /home/user/magic-pdf.json")
|
20 |
os.system("sed -i 's|cpu|cuda|g' /home/user/magic-pdf.json")
|
21 |
|
22 |
+
os.system('cp -r paddleocr /home/user/.paddleocr')
|
|
|
|
|
23 |
|
24 |
os.system("pip install gradio-pdf")
|
25 |
from gradio_pdf import PDF
|
|
|
152 |
model_manager = ModelSingleton()
|
153 |
txt_model = model_manager.get_model(False, False)
|
154 |
logger.info(f"txt_model init final")
|
155 |
+
# ocr_model = model_manager.get_model(True, False)
|
156 |
+
# logger.info(f"ocr_model init final")
|
157 |
return 0
|
158 |
except Exception as e:
|
159 |
logger.exception(e)
|
160 |
return -1
|
161 |
|
162 |
|
163 |
+
model_init = init_model()
|
164 |
+
logger.info(f"model_init: {model_init}")
|
165 |
|
166 |
|
167 |
if __name__ == "__main__":
|