Update app.py

app.py

@@ -74,7 +74,11 @@ if submit:
             
             input_column.markdown(f"<small>Compute Finished in {int(time.time() - last_time)} seconds</small>", unsafe_allow_html=True)
             prediction = np.argmax(result, axis=-1)
-            input_column.success(f"This news is {label[prediction]}.")
+            if prediction == 0:
+                input_column.success(f"This news is {label[prediction]}.")
+            else:
+                input_column.error(f"This news is {label[prediction]}.")
+                
             input_column.text(f"{int(result[prediction]*100)}% confidence")
             input_column.progress(result[prediction])