Butibol commited on
Commit
e1b9589
·
verified ·
1 Parent(s): 6e9244e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -2,24 +2,19 @@ import gradio as gr
2
  from fastai.vision.all import *
3
 
4
  # 1. Cargamos el modelo
5
- # Asegúrate de que tu archivo se llame 'model.pkl' o cambia el nombre aquí
6
  learn = load_learner('model.pkl')
7
 
8
- # 2. Definimos las categorías (lo que tu modelo sabe reconocer)
9
  categories = learn.dls.vocab
10
 
11
  def predict(img):
12
- # El modelo hace la predicción
13
  pred, idx, probs = learn.predict(img)
14
- # Devolvemos un diccionario con las probabilidades de cada categoría
15
  return dict(zip(categories, map(float, probs)))
16
 
17
  # 3. Configuramos la interfaz de Gradio
18
  image = gr.Image()
19
  label = gr.Label()
20
 
21
- # Aquí puedes poner imágenes de ejemplo si las subiste a tu Space
22
- examples = ['ejemplo1.jpg', 'ejemplo2.jpg']
23
-
24
- intf = gr.Interface(fn=predict, inputs=image, outputs=label, examples=examples)
25
  intf.launch(inline=False)
 
2
  from fastai.vision.all import *
3
 
4
  # 1. Cargamos el modelo
5
+ # Asegúrate de que el archivo se llame exactamente model.pkl en tu lista de archivos
6
  learn = load_learner('model.pkl')
7
 
8
+ # 2. Definimos las categorías
9
  categories = learn.dls.vocab
10
 
11
  def predict(img):
 
12
  pred, idx, probs = learn.predict(img)
 
13
  return dict(zip(categories, map(float, probs)))
14
 
15
  # 3. Configuramos la interfaz de Gradio
16
  image = gr.Image()
17
  label = gr.Label()
18
 
19
+ intf = gr.Interface(fn=predict, inputs=image, outputs=label)
 
 
 
20
  intf.launch(inline=False)