Hector Lopez commited on
Commit
284fe0a
·
1 Parent(s): db41ce0
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -39,6 +39,7 @@ def get_checkpoint(checkpoint_path):
39
  def predict(model, image):
40
  #img = PIL.Image.open(image)
41
  img = PIL.Image.open(BytesIO(image))
 
42
  print(img.shape)
43
  class_map = ClassMap(classes=['Waste'])
44
  transforms = tfms.A.Adapter([
 
39
  def predict(model, image):
40
  #img = PIL.Image.open(image)
41
  img = PIL.Image.open(BytesIO(image))
42
+ img = np.array(img)
43
  print(img.shape)
44
  class_map = ClassMap(classes=['Waste'])
45
  transforms = tfms.A.Adapter([