Update app.py
Browse files
app.py
CHANGED
|
@@ -27,4 +27,4 @@ data = st.text_area("Enter or paste a review")
|
|
| 27 |
if st.button('Predict'):
|
| 28 |
prediction = classify_text(data)
|
| 29 |
st.header(
|
| 30 |
-
f"{prediction}")
|
|
|
|
| 27 |
if st.button('Predict'):
|
| 28 |
prediction = classify_text(data)
|
| 29 |
st.header(
|
| 30 |
+
f"Negative Confidence: {prediction[0][0].item()}, Positive Confidence: {prediction[0][1].item()}")
|