Update app.py
Browse files
app.py
CHANGED
@@ -32,9 +32,9 @@ def transform_image(image):
|
|
32 |
mask = np.reshape(mask, (480, 640))
|
33 |
return Image.fromarray(mask.astype('uint8'))
|
34 |
|
35 |
-
#
|
36 |
interface = gr.Interface(fn=transform_image,
|
37 |
-
inputs=gr.components.Image(
|
38 |
outputs=gr.components.Image(),
|
39 |
examples=['color_154.jpg', 'color_189.jpg'])
|
40 |
-
interface.launch(share=False)
|
|
|
32 |
mask = np.reshape(mask, (480, 640))
|
33 |
return Image.fromarray(mask.astype('uint8'))
|
34 |
|
35 |
+
# Ajuste de la creación de la interfaz con los parámetros correctos para el componente Image
|
36 |
interface = gr.Interface(fn=transform_image,
|
37 |
+
inputs=gr.components.Image(width=640, height=480),
|
38 |
outputs=gr.components.Image(),
|
39 |
examples=['color_154.jpg', 'color_189.jpg'])
|
40 |
+
interface.launch(share=False)
|