Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,10 +7,10 @@ learn = load_learner("modeloFotos")
|
|
7 |
|
8 |
# Definimos una función que se encarga de llevar a cabo las predicciones
|
9 |
def predict(img):
|
10 |
-
|
11 |
img=np.array(learn.predict(img)[0]).reshape((128,128,3))
|
12 |
|
13 |
return Image.fromarray(img.astype("uint8"))
|
14 |
|
15 |
# Creamos la interfaz y la lanzamos.
|
16 |
-
gr.Interface(fn=predict,
|
|
|
7 |
|
8 |
# Definimos una función que se encarga de llevar a cabo las predicciones
|
9 |
def predict(img):
|
10 |
+
img = PILImage.create(img)
|
11 |
img=np.array(learn.predict(img)[0]).reshape((128,128,3))
|
12 |
|
13 |
return Image.fromarray(img.astype("uint8"))
|
14 |
|
15 |
# Creamos la interfaz y la lanzamos.
|
16 |
+
gr.Interface(fn=predict, inputs=gr.inputs.Image(type="filepath"), outputs=gr.outputs.Image(type="pil"),examples=['f1.jpg','f2.jpg']).launch(share=False)
|