Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|