Putting all in the app file

app.py

@@ -13,6 +13,9 @@ from config import HUGGINGFACEHUB_API_TOKEN
 
 from transformers import pipeline
 
+import gradio as gr
+
+
 os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
 
 # Vous pouvez choisir parmi les nombreux midèles disponibles sur HugginFace (https://huggingface.co/models)
@@ -86,3 +89,10 @@ def gradio_adapted_RAG(model_name, query):
     return generated_answer
 
 
+dropdown = gr.Dropdown(choices=["distilbert-base-uncased-distilled-squad",
+                                       "impira/layoutlm-document-qa", 
+                                       "impira/layoutlm-invoices"], label="Choose a model")
+
+iface = gr.Interface(fn=gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
+iface.launch()
+