File size: 980 Bytes
d748bf5
 
 
ac7b15a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d748bf5
ac7b15a
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os
os.system('pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.9/index.html')

from deepdoctection.dataflow import DataFromList
from deepdoctection import get_dd_analyzer
import gradio as gr


def analyze_image(img):
    # creating an image object and passing to the analyzer by using dataflows
    image = Image(file_name="input.png", location="")
    image.image = img

    df = DataFromList(lst=[image])

    analyzer = get_dd_analyzer(ocr=False)

    df = analyzer.analyze(dataset_dataflow=df)
    df.reset_state()
    dp = next(iter(df))


    return dp.viz(show_cells=False)

inputs = gr.inputs.Image(type='pil', label="Original Image")
outputs = gr.outputs.Image(type="pil", label="Output Image")

title = "Deepdoctection"
description = ""

# examples = [
#    ['example-table.jpeg'],
#    ['paper-image.jpeg']

# ]

gr.Interface(analyze_image, inputs, outputs, title=title, description=description, examples=examples).launch()