mirari commited on
Commit
2c158de
1 Parent(s): 8aa4824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ labels = learn.dls.vocab
7
 
8
  def predict(frase):
9
  pred, pred_idx, probs = learn.predict(frase)
10
- return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
  gr.Interface(fn=predict, inputs="text", outputs="text", examples=['im feeling rather rotten so im not very ambitious right now','im updating my blog because i feel shitty
13
  ']).launch(share=False)
 
7
 
8
  def predict(frase):
9
  pred, pred_idx, probs = learn.predict(frase)
10
+ return pred
11
 
12
  gr.Interface(fn=predict, inputs="text", outputs="text", examples=['im feeling rather rotten so im not very ambitious right now','im updating my blog because i feel shitty
13
  ']).launch(share=False)