janasumit2911 commited on
Commit
e81b67b
·
verified ·
1 Parent(s): 0d61eb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def predict_text(params):
49
  pred_label = tf.argmax(logits, axis=1).numpy()[0]
50
 
51
  label = {1: 'positive', 0: 'negative'}
52
- result = {'text': text, 'label': label[pred_label]}
53
  solutions.append(result)
54
 
55
  result_url = f"{api}/{job_id}"
 
49
  pred_label = tf.argmax(logits, axis=1).numpy()[0]
50
 
51
  label = {1: 'positive', 0: 'negative'}
52
+ result = {'text': text, 'label': [label[pred_label]]}
53
  solutions.append(result)
54
 
55
  result_url = f"{api}/{job_id}"