Kaushik066 commited on
Commit
d5ddb65
·
1 Parent(s): 9a563a7

from transformers import set_seed

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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