Commit
·
44d34ec
1
Parent(s):
caa24e0
import datasets
Browse files
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
|