mratanusarkar commited on
Commit
2691833
·
1 Parent(s): e6f2eb8

chore: address review points

Browse files
medrag_multi_modal/document_loader/image_loader/fitzpil_img_loader.py CHANGED
@@ -37,7 +37,7 @@ class FitzPILImageLoader(BaseImageLoader):
37
  loader.load_data(
38
  start_page=32,
39
  end_page=37,
40
- wandb_artifact_name="grays-anatomy-images",
41
  cleanup=False,
42
  )
43
  )
 
37
  loader.load_data(
38
  start_page=32,
39
  end_page=37,
40
+ wandb_artifact_name="grays-anatomy-images-fitzpil",
41
  cleanup=False,
42
  )
43
  )
medrag_multi_modal/document_loader/image_loader/marker_img_loader.py CHANGED
@@ -36,7 +36,7 @@ class MarkerImageLoader(BaseImageLoader):
36
  loader.load_data(
37
  start_page=31,
38
  end_page=36,
39
- wandb_artifact_name="grays-anatomy-images",
40
  cleanup=False,
41
  )
42
  )
 
36
  loader.load_data(
37
  start_page=31,
38
  end_page=36,
39
+ wandb_artifact_name="grays-anatomy-images-marker",
40
  cleanup=False,
41
  )
42
  )
medrag_multi_modal/document_loader/image_loader/pdf2image_img_loader.py CHANGED
@@ -36,7 +36,7 @@ class PDF2ImageLoader(BaseImageLoader):
36
  loader.load_data(
37
  start_page=31,
38
  end_page=36,
39
- wandb_artifact_name="grays-anatomy-images",
40
  cleanup=False,
41
  )
42
  )
 
36
  loader.load_data(
37
  start_page=31,
38
  end_page=36,
39
+ wandb_artifact_name="grays-anatomy-images-pdf2image",
40
  cleanup=False,
41
  )
42
  )
medrag_multi_modal/document_loader/image_loader/pdfplumber_img_loader.py CHANGED
@@ -35,7 +35,7 @@ class PDFPlumberImageLoader(BaseImageLoader):
35
  loader.load_data(
36
  start_page=32,
37
  end_page=37,
38
- wandb_artifact_name="grays-anatomy-images",
39
  cleanup=False,
40
  )
41
  )
 
35
  loader.load_data(
36
  start_page=32,
37
  end_page=37,
38
+ wandb_artifact_name="grays-anatomy-images-pdfplumber",
39
  cleanup=False,
40
  )
41
  )
medrag_multi_modal/document_loader/image_loader/pymupdf_img_loader.py CHANGED
@@ -37,7 +37,7 @@ class PyMuPDFImageLoader(BaseImageLoader):
37
  loader.load_data(
38
  start_page=32,
39
  end_page=37,
40
- wandb_artifact_name="grays-anatomy-images",
41
  cleanup=False,
42
  )
43
  )
 
37
  loader.load_data(
38
  start_page=32,
39
  end_page=37,
40
+ wandb_artifact_name="grays-anatomy-images-pymupdf",
41
  cleanup=False,
42
  )
43
  )