Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ labels= model.config.label2id
|
|
| 11 |
|
| 12 |
text = st.text_area("Enter some text!")
|
| 13 |
if text:
|
| 14 |
-
out = tf.math.softmax(model(tokenizer(raw_inputs, padding=True, truncation=True, return_tensors="np")).logits,
|
| 15 |
res = out.numpy()
|
| 16 |
|
| 17 |
labels['NEGATIVE'] = res[0,0]
|
|
|
|
| 11 |
|
| 12 |
text = st.text_area("Enter some text!")
|
| 13 |
if text:
|
| 14 |
+
out = tf.math.softmax(model(tokenizer(raw_inputs, padding=True, truncation=True, return_tensors="np")).logits, axis = -1)
|
| 15 |
res = out.numpy()
|
| 16 |
|
| 17 |
labels['NEGATIVE'] = res[0,0]
|