File size: 376 Bytes
ee7f4af
d799b27
ee7f4af
d799b27
 
 
ee7f4af
d799b27
ee7f4af
1
2
3
4
5
6
7
8
9
import gradio as gr
import RAG

dropdown = gr.Dropdown(choices=["distilbert-base-uncased-distilled-squad",
                                       "impira/layoutlm-document-qa", 
                                       "impira/layoutlm-invoices"], label="Choose a model")

iface = gr.Interface(fn=RAG.gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
iface.launch()