Jupeream commited on
Commit
5c188bc
1 Parent(s): 26bb756

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  learn = load_learner('export (2).pkl')
7
 
8
  # Definimos las etiquetas de nuestro modelo
9
- labels = learn.dls.vocab
10
 
11
 
12
  # Definimos una función que se encarga de llevar a cabo las predicciones
@@ -15,5 +15,5 @@ def predict(sentence):
15
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
16
 
17
  # Creamos la interfaz y la lanzamos.
18
- gr.Interface(fn=predict, inputs="text", outputs=gr.outputs.Label(num_top_classes=3)).launch(share=False)
19
 
 
6
  learn = load_learner('export (2).pkl')
7
 
8
  # Definimos las etiquetas de nuestro modelo
9
+ labels = ['1','2','3','4', '5']
10
 
11
 
12
  # Definimos una función que se encarga de llevar a cabo las predicciones
 
15
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
16
 
17
  # Creamos la interfaz y la lanzamos.
18
+ gr.Interface(fn=predict, inputs=gr.inputs.Textbox(lines=1), outputs=gr.outputs.Label(num_top_classes=3), examples=['I feel so happy today','i feel selfish and spoiled']).launch(share=False)
19