import gradio as gr
from demo import predict

gr.Interface(fn=predict, 
            inputs=[gr.Image(label = 'You can select an example to quickly submit'), "text"], 
            outputs=["text"],
            examples=[
            ['PDF_val_151.png','How many table objects are located at the top side of figure?'],
            ['PDF_val_90.png', 'Where is the caption of the figure located at?'],
            ['PDF_val_64.png','How many text objects are located at the bottom side of figure?'],
            ['PDF_val_26.png','Are there any title exist?'],
            ['PDF_val_60.png','Where is the caption of the table located at?'],
            ['PDF_val_158.png','Does title objects exist in this page?']],
            title = 'V-Doc : Visual questions answers with Documents',
            description = 'Paper: https://arxiv.org/abs/2205.13724').launch()

if __name__ == '__main__':
    io.lanuch()