Update app.py

app.py

@@ -49,7 +49,7 @@ with st.form(key='input_form'):
 if button:
     if to_analyze:
         pred = make_prediction(to_analyze)
-        st.markdown(pred)
+        st.markdown("Negative" if torch.argmax(pred.logits).item() == 0 else "Positive")
     else:
         st.markdown("Empty request. Please resubmit")