sergio-sanz-rodriguez commited on
Commit
4df6cc7
·
1 Parent(s): b64811b

fixed bug model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -1,4 +1,5 @@
1
  import torch
 
2
  from vision_transformer import ViT
3
  from torchvision.transforms import v2
4
 
 
1
  import torch
2
+ from pathlib import Path
3
  from vision_transformer import ViT
4
  from torchvision.transforms import v2
5