Spaces:
Sleeping
Sleeping
Commit
·
45b79c0
1
Parent(s):
e34360f
update
Browse files
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 |
-
|
| 72 |
-
|
| 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=[
|