Namitg02 commited on
Commit
cb97c30
·
verified ·
1 Parent(s): 7db9774

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -51,6 +51,13 @@ map_style_dataset = MapStyleDataset(iterable)
51
  # Create a DataLoader for the MapStyle dataset
52
  data_loader = torch.utils.data.DataLoader(map_style_dataset, batch_size=2)
53
 
 
 
 
 
 
 
 
54
 
55
  #datasetiter = load_dataset("Namitg02/Test", split='train', streaming=False)
56
  #dataset = to_map_style_dataset(datasetiter)
 
51
  # Create a DataLoader for the MapStyle dataset
52
  data_loader = torch.utils.data.DataLoader(map_style_dataset, batch_size=2)
53
 
54
+ if is_iterable_dataset(map_style_dataset):
55
+ print("The dataset is iterable-style.")
56
+ else:
57
+ print("The dataset is map-style.")
58
+
59
+
60
+
61
 
62
  #datasetiter = load_dataset("Namitg02/Test", split='train', streaming=False)
63
  #dataset = to_map_style_dataset(datasetiter)