dolo650/QA_my_PDF
1
1o
2 ��Ke� � @ s� d dl Zd dlmZ d dlmZ d dlmZ d dlm Z e e3�j Z eddd�Z
d ed4ed efd d
�Zejeejdd�edd�ge�� dee d �ggd�Ze�� dS )� N)�PDF)�convert_from_path)�pipeline)�Pathzdocument-question-answeringzimpira/layoutlm-document-qa)�model�doc�question�returnc C s0 t | �d }t||�}t|dd� dd�d d S )Nr c S s | d S )N�score� )�xr r �0/Users/freddy/sources/components/pdf/demo/app.py�<lambda> s zqa.<locals>.<lambda>T)�key�reverse�answer)r �p�sorted)r r �img�outputr r r
�qa s 5r �Question)�label�Documentz What is the total gross worth?z
invoice_2.pdf)�examples)�gradio�gr�6gradio_pdfr � pdf2imager � transformersr �pathlibr �__file__�parent�dir_r �strr � Interface�Textbox�demo�launchr r r r
�<module> s$ 7��