edgilr commited on
Commit
1e03efb
1 Parent(s): 26ccbd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def predict(txt):
18
  #pred,pred_idx,probs = learner.predict(txt)
19
  #return pred
20
  # Modelo de clasificación basados en mecanismos de atención
21
- return classifier(txt)
22
 
23
  gr.Interface(fn=predict, inputs="text", outputs="text",
24
  examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',
 
18
  #pred,pred_idx,probs = learner.predict(txt)
19
  #return pred
20
  # Modelo de clasificación basados en mecanismos de atención
21
+ return classifier(txt)[0]['label']
22
 
23
  gr.Interface(fn=predict, inputs="text", outputs="text",
24
  examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',