Update app.py

app.py

@@ -14,24 +14,20 @@ login(token=HF_TOKEN, add_to_git_credential=True)
 
 
 def stream_dataset_from_hub(split):
-    dataset = load_dataset('mcarthuradal/arm-unicef',
-                           data_dir='images',
-                           split=split,
-                           streaming=True)
-    return dataset
+    dataset = load_dataset_builder('mcarthuradal/arm-unicef')
+    data = dataset.as_streaming_dataset(split).iter(200)
+    yield next(data)
 
 
-dataset = stream_dataset_from_hub('train')
+stream = stream_dataset_from_hub('train')
 
 
-def get_images(path: str | Path)  :
-    # filenames = os.listdir(path)
-
-    # image_names = [filename for filename in filenames if filename.endswith('.tif')]
+def get_images(split: str):
+    
     n = 50
-    image_batch = dataset.iter(n)
+    batch = stream['image'][:n]
 
-    yield image_batch
+    return  batch
 
 
 iface = gr.Interface(fn=get_images,
@@ -48,4 +44,5 @@ gr.Gallery(columns=5,
            show_download_button=False,
            show_share_button=False)
 
-iface.launch(inbrowser=True)
+iface.launch(debug=True)
+