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