jeysshon commited on
Commit
51339d2
·
verified ·
1 Parent(s): 8d006e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -5,20 +5,21 @@ import skimage
5
  learn = load_learner('model.pkl')
6
 
7
  labels = learn.dls.vocab
 
8
  def predict(img):
9
  img = PILImage.create(img)
10
  pred, pred_idx, probs = learn.predict(img)
11
  return pred
12
 
13
- title = "Modelo de Predicción Aves PARCIAL PIB "
14
  description = "Este modelo utiliza redes neuronales para clasificar diferentes especies de aves en imágenes. En la parte inferior, puedes probar el modelo con algunas imágenes de muestra."
15
  examples = ['3.jpg', '5.jpg', 'images.jpeg', '1.jpg', '2.jpg', '4.jpg', '6.jpg']
16
  enable_queue = True
17
 
18
  gr.Interface(
19
  fn=predict,
20
- inputs=gr.inputs.Image(shape=(224, 224)),
21
- outputs=gr.outputs.Label(num_top_classes=1),
22
  title=title,
23
  description=description,
24
  examples=examples,
 
5
  learn = load_learner('model.pkl')
6
 
7
  labels = learn.dls.vocab
8
+
9
  def predict(img):
10
  img = PILImage.create(img)
11
  pred, pred_idx, probs = learn.predict(img)
12
  return pred
13
 
14
+ title = "Modelo de Predicción Aves PARCIAL PIB"
15
  description = "Este modelo utiliza redes neuronales para clasificar diferentes especies de aves en imágenes. En la parte inferior, puedes probar el modelo con algunas imágenes de muestra."
16
  examples = ['3.jpg', '5.jpg', 'images.jpeg', '1.jpg', '2.jpg', '4.jpg', '6.jpg']
17
  enable_queue = True
18
 
19
  gr.Interface(
20
  fn=predict,
21
+ inputs=gr.Image(shape=(224, 224)),
22
+ outputs=gr.Label(num_top_classes=1),
23
  title=title,
24
  description=description,
25
  examples=examples,