Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,15 +1,18 @@
|
|
| 1 |
import os
|
| 2 |
os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
|
| 3 |
|
| 4 |
-
from deepdoctection.dataflow import DataFromList
|
| 5 |
-
from deepdoctection import get_dd_analyzer
|
| 6 |
import deepdoctection as dd
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
import gradio as gr
|
| 8 |
|
| 9 |
_DD_ONE = "deepdoctection/configs/conf_dd_one.yaml"
|
| 10 |
_TESSERACT = "deepdoctection/configs/conf_tesseract.yaml"
|
| 11 |
|
| 12 |
-
dd.ModelCatalog.register("layout/model_final_inf_only.pt",
|
| 13 |
name="layout/model_final_inf_only.pt",
|
| 14 |
description="Detectron2 layout detection model trained on private datasets",
|
| 15 |
config="dd/d2/layout/CASCADE_RCNN_R_50_FPN_GN.yaml",
|
|
|
|
| 1 |
import os
|
| 2 |
os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')
|
| 3 |
|
|
|
|
|
|
|
| 4 |
import deepdoctection as dd
|
| 5 |
+
|
| 6 |
+
from deepdoctection.extern.model import ModelProfile
|
| 7 |
+
from deepdoctection.analyzer.dd import build_analyzer, _auto_select_lib_and_device, _maybe_copy_config_to_cache
|
| 8 |
+
from deepdoctection.utils.metacfg import set_config_by_yaml
|
| 9 |
+
|
| 10 |
import gradio as gr
|
| 11 |
|
| 12 |
_DD_ONE = "deepdoctection/configs/conf_dd_one.yaml"
|
| 13 |
_TESSERACT = "deepdoctection/configs/conf_tesseract.yaml"
|
| 14 |
|
| 15 |
+
dd.ModelCatalog.register("layout/model_final_inf_only.pt",ModelProfile(
|
| 16 |
name="layout/model_final_inf_only.pt",
|
| 17 |
description="Detectron2 layout detection model trained on private datasets",
|
| 18 |
config="dd/d2/layout/CASCADE_RCNN_R_50_FPN_GN.yaml",
|