Update app.py

app.py

@@ -215,7 +215,8 @@ def load_reviews_from_file(file_path):
 
 #Demo Section
 st.title("SpaGAN Demo: IN PROGRESS")
-st.write("This demo allows you to select from a list of sample reviews that contrain both real and fake reviews. This demo will highlight all the entity types found within the review and display the prediction of the model.")
+st.write("This demo allows you to select from a list of sample reviews that contrain both real and fake reviews. 
+        This demo will highlight all the entity types found within the review and display the prediction of the model.")
 
 # Define a color map and descriptions for different entity types
 COLOR_MAP = {
@@ -253,8 +254,14 @@ if st.button("Highlight Geo-Entities"):
         st.write("Concatenated Embedding:", combined_embedding)
 
         prediction = get_prediction(combined_embedding)
-        st.write("Prediction:", prediction)
         
+        if(prediction == 0):
+            st.write("Prediction: Not Spam")
+        else if(prediction == 1):
+            st.write("Prediction: Spam")
+        else:
+            st.write("error during prediction")
+            
         # Process the text using spaCy
         doc = nlp(selected_review)