luisvarona commited on
Commit
04e2419
·
verified ·
1 Parent(s): 10f8043

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -80,7 +80,8 @@ model = model.cpu()
80
 
81
  # Definimos una función que se encarga de llevar a cabo las predicciones
82
  def predict(img_ruta):
83
- img = PIL.Image.open(img_ruta)
 
84
  image = transforms.Resize((480,640))(img)
85
  tensor = transform_image(image=image)
86
  model.to(device)
 
80
 
81
  # Definimos una función que se encarga de llevar a cabo las predicciones
82
  def predict(img_ruta):
83
+ # img = PIL.Image.open(img_ruta)
84
+ img = img_ruta
85
  image = transforms.Resize((480,640))(img)
86
  tensor = transform_image(image=image)
87
  model.to(device)