Jkalonji commited on
Commit
3fa7f0c
1 Parent(s): 76a766c

adding math models

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -88,12 +88,11 @@ def gradio_adapted_RAG(model_name, query):
88
  return generated_answer
89
 
90
 
91
- dropdown = gr.Dropdown(choices=["distilbert-base-uncased-distilled-squad",
92
- "sentence-transformers/all-MiniLM-L6-v2",
93
- "deepset/roberta-base-squad2",
94
- "CATIE-AQ/QAmembert-large",
95
- "FlagAlpha/Llama2-Chinese-13b-Chat",
96
- "microsoft/orca-math-word-problems-200k"], label="Choose a model")
97
 
98
  iface = gr.Interface(fn=gradio_adapted_RAG, inputs=[dropdown, "text"], outputs="text")
99
  iface.launch()
 
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()