Spaces:
Sleeping
Sleeping
luisvarona
commited on
Commit
•
2f0d4f4
1
Parent(s):
96896c5
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,8 @@ model = model.cpu()
|
|
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)
|
@@ -101,4 +102,4 @@ def predict(img_ruta):
|
|
101 |
#img = PILImage.create(img) #igual hay que usar esto en vez de PIL.Image.open
|
102 |
|
103 |
# Creamos la interfaz y la lanzamos.
|
104 |
-
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(480, 640)), outputs=gr.inputs.Image(shape=(480, 640))
|
|
|
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 |
+
img = PIL.Image.fromarray(img_ruta)
|
86 |
image = transforms.Resize((480,640))(img)
|
87 |
tensor = transform_image(image=image)
|
88 |
model.to(device)
|
|
|
102 |
#img = PILImage.create(img) #igual hay que usar esto en vez de PIL.Image.open
|
103 |
|
104 |
# Creamos la interfaz y la lanzamos.
|
105 |
+
gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(480, 640)), outputs=gr.inputs.Image(shape=(480, 640)), examples=['color_155.jpg','color_154 (1).jpg']).launch(share=False)
|