Mawube commited on
Commit
a43c7ca
·
verified ·
1 Parent(s): 3b27d39

Return image as a list

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -28,7 +28,7 @@ def preprocess_image(image_path):
28
  transform = get_transform()
29
  img = transform(img)
30
 
31
- return img
32
 
33
 
34
  def get_transform():
 
28
  transform = get_transform()
29
  img = transform(img)
30
 
31
+ return [img]
32
 
33
 
34
  def get_transform():