| 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() |