Update app.py

app.py

@@ -59,13 +59,14 @@ if st.button('Load Model', disabled=False):
                 prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
                 pred = 'Predicted Class: '+ prediction
                 return pred
-        placeholder.text_input("Enter the text you'd like to analyze for spam.", disabled=False, key="2")
-        placeholder2.button('Analyze', disabled=False, key="2")       
-if text or aButton:
-    placeholder.text_input("Enter the text you'd like to analyze for spam.", disabled=False, key="3")
-    placeholder2.button('Analyze', disabled=False, key="3") 
-    with col2:
-        st.header(predict(text))
+            with col2:
+                placeholder.text_input("Enter the text you'd like to analyze for spam.", disabled=False, key="2")
+                placeholder2.button('Analyze', disabled=False, key="2")       
+            if text or aButton:
+                placeholder.text_input("Enter the text you'd like to analyze for spam.", disabled=False, key="3")
+                placeholder2.button('Analyze', disabled=False, key="3") 
+                with col2:
+                    st.header(predict(text))