Spaces:
Running
Running
geekyrakshit
commited on
Commit
•
716efd2
1
Parent(s):
170d9a9
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(
|