Commit
·
d5ddb65
1
Parent(s):
9a563a7
from transformers import set_seed
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ from torchvision import transforms
|
|
4 |
# For ML Model
|
5 |
import transformers
|
6 |
from transformers import VivitImageProcessor, VivitConfig, VivitModel
|
|
|
|
|
7 |
# For Data Loaders
|
8 |
import datasets
|
9 |
from torch.utils.data import Dataset, DataLoader
|
|
|
4 |
# For ML Model
|
5 |
import transformers
|
6 |
from transformers import VivitImageProcessor, VivitConfig, VivitModel
|
7 |
+
from transformers import set_seed
|
8 |
+
|
9 |
# For Data Loaders
|
10 |
import datasets
|
11 |
from torch.utils.data import Dataset, DataLoader
|