gabgrenier commited on
Commit
7aed866
·
1 Parent(s): 38b8d2f

added model

Browse files
Files changed (1) hide show
  1. test.py +1 -0
test.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torchvision.transforms.functional as tf
2
  from harmonizer.src import model
3
  harmonizer = model.Harmonizer()
 
1
+ import torch
2
  import torchvision.transforms.functional as tf
3
  from harmonizer.src import model
4
  harmonizer = model.Harmonizer()