Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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)
|