test new req

helper/examples/examples.py

@@ -14,8 +14,8 @@ class DemoImages:
 
     def __init__(self, url="Riksarkivet/test_images_demo", cache_dir="./helper/examples/.cache_images"):
         if not hasattr(self, "images_datasets"):
-            self.images_datasets = datasets.load_dataset(url, cache_dir=cache_dir)
-            self.example_df = self.images_datasets["train"].to_pandas()
+            self.images_datasets = datasets.load_dataset(url, cache_dir=cache_dir, split="train")
+            self.example_df = self.images_datasets.to_pandas()
             self.examples_list = self.convert_bytes_to_images()
 
     def convert_bytes_to_images(self):
@@ -41,6 +41,9 @@ class DemoImages:
 
 
 if __name__ == "__main__":
-    test = DemoImages(cache_dir=".cache_images")
+    # test = DemoImages(cache_dir=".cache_images")
 
-    print(test.examples_list)
+    # print(test.examples_list)
+
+    images_datasets = datasets.load_dataset("Riksarkivet/test_images_demo", cache_dir="./helper/examples/.cache_images")
+    print(images_datasets["train"]["image"][0])

requirements.txt

@@ -1,14 +1,14 @@
 torch==2.0.1
 torchvision==0.15.2
 openmim==0.3.9
-gradio==3.41.1
+gradio==3.41.0
 pandas==2.0.3
 numpy==1.25.0
 opencv-python-headless==4.7.0.72
 Jinja2==3.1.2
 transformers==4.30.2
 huggingface-hub==0.15.1
-datasets==2.13.1
+datasets==2.14.5
 requests==2.31.0
 pillow==9.5.0
 
@@ -19,4 +19,5 @@ pillow==9.5.0
 # !mim install mmengine
 # !mim install mmcv
 # !mim install mmdet
-# !mim install mmocr
+# !mim install mmocr
+