minor bug fix in output

app.py

@@ -80,8 +80,8 @@ if st.button("Submit"):
     st.title("Predicted Patentability")
     if probability[0][0] > probability[0][1]:
         st.write("Rejection Score:")
-        st.write(probability[0][0])
+        st.write(probability[0][0].item())
     else:
         st.write("Acceptance Score:")
-        st.write(probability[0][1])
-    st.write(pred_label)
+        st.write(probability[0][1].item())
+    st.write("Result:", pred_label)