Trying out some models

app.py

@@ -89,8 +89,9 @@ def gradio_adapted_RAG(model_name, query):
 
 
 dropdown = gr.Dropdown(choices=["distilbert-base-uncased-distilled-squad",
-                                       "impira/layoutlm-document-qa", 
-                                       "impira/layoutlm-invoices"], label="Choose a model")
+                                       "deepset/roberta-base-squad2", 
+                                       "CATIE-AQ/QAmembert-large",
+                               "FlagAlpha/Llama2-Chinese-13b-Chat"], label="Choose a model")
 
 iface = gr.Interface(fn=gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
 iface.launch()