pminervini commited on
Commit
45b79c0
·
1 Parent(s): e34360f
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -68,10 +68,8 @@ def rag_pipeline(prompt, index="pubmed", num_docs=3, model_name="HuggingFaceH4/z
68
  iface = gr.Interface(fn=rag_pipeline,
69
  inputs=[
70
  gr.Textbox(label="Input Prompt", value="Are group 2 innate lymphoid cells (ILC2s) increased in chronic rhinosinusitis with nasal polyps or eosinophilia?"),
71
- gr.Column([
72
- gr.Dropdown(label="Index", choices=["pubmed", "wikipedia", "textbooks"], value="pubmed"),
73
- gr.Number(label="Number of Documents", value=3, step=1, minimum=1, maximum=10)
74
- ]),
75
  gr.Dropdown(label="Model", choices=["HuggingFaceH4/zephyr-7b-beta", "meta-llama/Llama-2-7b-chat-hf", "meta-llama/Llama-2-13b-chat-hf", "meta-llama/Llama-2-70b-chat-hf"], value="HuggingFaceH4/zephyr-7b-beta")
76
  ],
77
  outputs=[
 
68
  iface = gr.Interface(fn=rag_pipeline,
69
  inputs=[
70
  gr.Textbox(label="Input Prompt", value="Are group 2 innate lymphoid cells (ILC2s) increased in chronic rhinosinusitis with nasal polyps or eosinophilia?"),
71
+ gr.Dropdown(label="Index", choices=["pubmed", "wikipedia", "textbooks"], value="pubmed"),
72
+ gr.Number(label="Number of Documents", value=3, step=1, minimum=1, maximum=10),
 
 
73
  gr.Dropdown(label="Model", choices=["HuggingFaceH4/zephyr-7b-beta", "meta-llama/Llama-2-7b-chat-hf", "meta-llama/Llama-2-13b-chat-hf", "meta-llama/Llama-2-70b-chat-hf"], value="HuggingFaceH4/zephyr-7b-beta")
74
  ],
75
  outputs=[