Spaces:
Running
Running
update: app.py
Browse files
app.py
CHANGED
|
@@ -72,17 +72,17 @@ if retriever_type != "":
|
|
| 72 |
elif retriever_type == "Contriever":
|
| 73 |
retriever = ContrieverRetriever.from_index(
|
| 74 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-contriever",
|
| 75 |
-
|
| 76 |
)
|
| 77 |
elif retriever_type == "MedCPT":
|
| 78 |
retriever = MedCPTRetriever.from_index(
|
| 79 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-medcpt",
|
| 80 |
-
|
| 81 |
)
|
| 82 |
elif retriever_type == "NV-Embed-v2":
|
| 83 |
retriever = NVEmbed2Retriever.from_index(
|
| 84 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-nv-embed-2",
|
| 85 |
-
|
| 86 |
)
|
| 87 |
|
| 88 |
medqa_assistant = MedQAAssistant(
|
|
|
|
| 72 |
elif retriever_type == "Contriever":
|
| 73 |
retriever = ContrieverRetriever.from_index(
|
| 74 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-contriever",
|
| 75 |
+
chunk_dataset=chunk_dataset_id,
|
| 76 |
)
|
| 77 |
elif retriever_type == "MedCPT":
|
| 78 |
retriever = MedCPTRetriever.from_index(
|
| 79 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-medcpt",
|
| 80 |
+
chunk_dataset=chunk_dataset_id,
|
| 81 |
)
|
| 82 |
elif retriever_type == "NV-Embed-v2":
|
| 83 |
retriever = NVEmbed2Retriever.from_index(
|
| 84 |
index_repo_id="ashwiniai/medrag-text-corpus-chunks-nv-embed-2",
|
| 85 |
+
chunk_dataset=chunk_dataset_id,
|
| 86 |
)
|
| 87 |
|
| 88 |
medqa_assistant = MedQAAssistant(
|