paascorb commited on
Commit
c8f5bbf
1 Parent(s): a87211a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ infer_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(384),tfms.A.Normalize()])
17
 
18
  # Definimos una función que se encarga de llevar a cabo las predicciones
19
  def predict(img):
20
- img = PIL.Image.open(img)
21
  pred_dict = model_type(img, infer_tfms, model.to("cpu"), class_map=ClassMap(['raccoon']), detection_threshold=0.5)
22
  return pred_dict["img"]
23
 
 
17
 
18
  # Definimos una función que se encarga de llevar a cabo las predicciones
19
  def predict(img):
20
+ img = PIL.Image.fromarray(img, "RGB")
21
  pred_dict = model_type(img, infer_tfms, model.to("cpu"), class_map=ClassMap(['raccoon']), detection_threshold=0.5)
22
  return pred_dict["img"]
23