Spaces:
Build error
Build error
Update app.py
Browse files
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}"
|