getting back to old models
Browse files
app.py
CHANGED
@@ -88,11 +88,9 @@ def gradio_adapted_RAG(model_name, query):
|
|
88 |
return generated_answer
|
89 |
|
90 |
|
91 |
-
dropdown = gr.Dropdown(choices=["
|
92 |
-
|
93 |
-
|
94 |
-
"deepseek-ai/deepseek-math-7b-instruct",
|
95 |
-
"MaziyarPanahi/WizardLM-Math-70B-v0.1"], label="Choose a model")
|
96 |
|
97 |
iface = gr.Interface(fn=gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
|
98 |
iface.launch()
|
|
|
88 |
return generated_answer
|
89 |
|
90 |
|
91 |
+
dropdown = gr.Dropdown(choices=["distilbert-base-uncased-distilled-squad",
|
92 |
+
"impira/layoutlm-document-qa",
|
93 |
+
"impira/layoutlm-invoices"], label="Choose a model")
|
|
|
|
|
94 |
|
95 |
iface = gr.Interface(fn=gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
|
96 |
iface.launch()
|