giorgio-caparvi commited on
Commit
f687747
·
1 Parent(s): a6b8e2c

fixed import

Browse files
api/model/src/datasets/vitonhd.py CHANGED
@@ -6,6 +6,7 @@ import os
6
  import pathlib
7
  import random
8
  import sys
 
9
  from typing import Tuple
10
 
11
  PROJECT_ROOT = pathlib.Path(__file__).absolute().parents[2].absolute()
 
6
  import pathlib
7
  import random
8
  import sys
9
+ import io
10
  from typing import Tuple
11
 
12
  PROJECT_ROOT = pathlib.Path(__file__).absolute().parents[2].absolute()
api/model/src/eval.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
 
3
  # external libraries
4
  import torch
 
1
  import os
2
+ import io
3
 
4
  # external libraries
5
  import torch