Spaces:

geekyrakshit
/

medrag

Running

mratanusarkar commited on Oct 20, 2024

Commit

2691833

1 Parent(s): e6f2eb8

chore: address review points

Files changed (5) hide show

medrag_multi_modal/document_loader/image_loader/fitzpil_img_loader.py CHANGED Viewed

@@ -37,7 +37,7 @@ class FitzPILImageLoader(BaseImageLoader):
             loader.load_data(
                 start_page=32,
                 end_page=37,
-                wandb_artifact_name="grays-anatomy-images",
                 cleanup=False,
             )
         )

             loader.load_data(
                 start_page=32,
                 end_page=37,
+                wandb_artifact_name="grays-anatomy-images-fitzpil",
                 cleanup=False,
             )
         )

medrag_multi_modal/document_loader/image_loader/marker_img_loader.py CHANGED Viewed

@@ -36,7 +36,7 @@ class MarkerImageLoader(BaseImageLoader):
             loader.load_data(
                 start_page=31,
                 end_page=36,
-                wandb_artifact_name="grays-anatomy-images",
                 cleanup=False,
             )
         )

             loader.load_data(
                 start_page=31,
                 end_page=36,
+                wandb_artifact_name="grays-anatomy-images-marker",
                 cleanup=False,
             )
         )

medrag_multi_modal/document_loader/image_loader/pdf2image_img_loader.py CHANGED Viewed

@@ -36,7 +36,7 @@ class PDF2ImageLoader(BaseImageLoader):
             loader.load_data(
                 start_page=31,
                 end_page=36,
-                wandb_artifact_name="grays-anatomy-images",
                 cleanup=False,
             )
         )

             loader.load_data(
                 start_page=31,
                 end_page=36,
+                wandb_artifact_name="grays-anatomy-images-pdf2image",
                 cleanup=False,
             )
         )

medrag_multi_modal/document_loader/image_loader/pdfplumber_img_loader.py CHANGED Viewed

@@ -35,7 +35,7 @@ class PDFPlumberImageLoader(BaseImageLoader):
             loader.load_data(
                 start_page=32,
                 end_page=37,
-                wandb_artifact_name="grays-anatomy-images",
                 cleanup=False,
             )
         )

             loader.load_data(
                 start_page=32,
                 end_page=37,
+                wandb_artifact_name="grays-anatomy-images-pdfplumber",
                 cleanup=False,
             )
         )

medrag_multi_modal/document_loader/image_loader/pymupdf_img_loader.py CHANGED Viewed

@@ -37,7 +37,7 @@ class PyMuPDFImageLoader(BaseImageLoader):
             loader.load_data(
                 start_page=32,
                 end_page=37,
-                wandb_artifact_name="grays-anatomy-images",
                 cleanup=False,
             )
         )

             loader.load_data(
                 start_page=32,
                 end_page=37,
+                wandb_artifact_name="grays-anatomy-images-pymupdf",
                 cleanup=False,
             )
         )