sacamiso commited on
Commit
c31089d
·
1 Parent(s): 4543349

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- #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(shape=(128, 128)), outputs=gr.outputs.Image(),examples=['f1.jpg','f2.jpg']).launch(share=False)
 
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)