Ubuntu commited on
Commit
2911751
·
1 Parent(s): 92d5e92
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +1 -1
  2. app.py +1 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -10,7 +10,7 @@ from pathlib import Path
10
  import pytorch_lightning as pl
11
  from torch.utils.data import DataLoader
12
  from torchmetrics import Accuracy
13
-
14
  HF_DATASETS_CACHE="./"
15
 
16
 
 
10
  import pytorch_lightning as pl
11
  from torch.utils.data import DataLoader
12
  from torchmetrics import Accuracy
13
+ from torchvision import transforms
14
  HF_DATASETS_CACHE="./"
15
 
16
 
app.py CHANGED
@@ -10,7 +10,7 @@ from pathlib import Path
10
  import pytorch_lightning as pl
11
  from torch.utils.data import DataLoader
12
  from torchmetrics import Accuracy
13
-
14
  HF_DATASETS_CACHE="./"
15
 
16
 
 
10
  import pytorch_lightning as pl
11
  from torch.utils.data import DataLoader
12
  from torchmetrics import Accuracy
13
+ from torchvision import transforms
14
  HF_DATASETS_CACHE="./"
15
 
16