geekyrakshit commited on
Commit
716efd2
1 Parent(s): 170d9a9

update: app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- chunk_dataset_id=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_id=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_id=chunk_dataset_id,
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(