Kaushik066 commited on
Commit
44d34ec
·
1 Parent(s): caa24e0

import datasets

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from torchvision import transforms
4
  # For ML Model
5
  from transformers import VivitImageProcessor, VivitConfig, VivitModel
6
  # For Data Loaders
 
7
  from torch.utils.data import Dataset, DataLoader
8
  # For GPU
9
  from accelerate import Accelerator, notebook_launcher
 
4
  # For ML Model
5
  from transformers import VivitImageProcessor, VivitConfig, VivitModel
6
  # For Data Loaders
7
+ import datasets
8
  from torch.utils.data import Dataset, DataLoader
9
  # For GPU
10
  from accelerate import Accelerator, notebook_launcher