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