Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 from keras import Sequential
-from keras.layers import Dense,Embedding
+from keras.layers import Dense,Embedding,Flatten
 from keras.utils import pad_sequences
 from keras.preprocessing.text import Tokenizer
 st.title("Spam-NonSpam Detector")
@@ -22,6 +22,7 @@ if st.button("Check"):
     voc_size=len(tokenizer.word_index)
     model = Sequential()
     model.add(Embedding(voc_size+1,2,input_length=61))
+    model.add(Flatten())
     model.add(Dense(5,activation="relu"))
     model.add(Dense(5,activation="relu"))
     model.add(Dense(1, activation='sigmoid'))
@@ -35,11 +36,11 @@ if st.button("Check"):
     seq=tokenizer.texts_to_sequences(Input)
     inp=pad_sequences(seq,padding='post',maxlen=61)
     a=model.predict(inp)
-    value=a.argmax()
+    value=a[0][0]
     st.text("Input:")
     st.markdown(Input[0])
     st.text("Output:")
-    if (value==1):
+    if (value>0.5):
       st.text('Non-spam message')
     else:
       st.text('Spam message')