Update app.py

app.py

@@ -51,6 +51,12 @@ map_style_dataset = MapStyleDataset(iterable)
 # Create a DataLoader for the MapStyle dataset
 data_loader = torch.utils.data.DataLoader(map_style_dataset, batch_size=2)
 
+def is_iterable_dataset(map_style_dataset):
+  return isinstance(map_style_dataset, torch.utils.data.IterableDataset)
+
+def is_map_style_dataset(map_style_dataset):
+  return isinstance(map_style_dataset, torch.utils.data.Dataset)
+
 if is_iterable_dataset(map_style_dataset):
   print("The dataset is iterable-style.")
 else: