Jkalonji commited on
Commit
56b46a1
1 Parent(s): 3fa7f0c

getting back to old models

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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=["tbs17/MathBERT",
92
- "witiko/mathberta",
93
- "meta-math/MetaMath-Mistral-7B",
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()