luisvarona commited on
Commit
ca98dbf
·
verified ·
1 Parent(s): 0a88686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,8 +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) #esto si el parámetro de entrada es una ruta a una imagen
84
- # img = img_ruta # esto si el parámetro de entrada es una imagen
85
  image = transforms.Resize((480,640))(img)
86
  tensor = transform_image(image=image)
87
  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) #esto si el parámetro de entrada es una ruta a una imagen
84
+ img = img_ruta # esto si el parámetro de entrada es una imagen
85
  image = transforms.Resize((480,640))(img)
86
  tensor = transform_image(image=image)
87
  model.to(device)