ancebuc commited on
Commit
8d02e1a
·
1 Parent(s): 26ec02b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ labels = [i for i in range(6)]
11
  def predict(text):
12
  #img = PILImage.create(img)
13
  pred = learner.predict(text)
14
- return pred
15
 
16
  # Creamos la interfaz y la lanzamos.
17
  gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label()).launch(share=False)
 
11
  def predict(text):
12
  #img = PILImage.create(img)
13
  pred = learner.predict(text)
14
+ return pred[0]
15
 
16
  # Creamos la interfaz y la lanzamos.
17
  gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label()).launch(share=False)