Update app.py

app.py

@@ -43,9 +43,8 @@ with col2:
             
             device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
             #Used for printing the name if the variables. Removing it will not intrupt the project.
-            def namestr(obj, namespace):
-                return [name for name in namespace if namespace[name] is obj]
-            
+
+            st.success("Model Loaded!")
             def predict(new_sentence):
                 # We need Token IDs and Attention Mask for inference on the new sentence
                 test_ids = []
@@ -66,11 +65,11 @@ with col2:
             
                 prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
                 pred = 'Predicted Class: '+ prediction
+                with col2:
+                    st.header(pred)
+                    text = st.text_input("Enter the text you'd like to analyze for spam.")
+                    if text or st.button('Analyze'):
+                        predict(text)
                 
-        st.success("Model Loaded!")
         
-with col2:
-    st.header(pred)
-    text = st.text_input("Enter the text you'd like to analyze for spam.")
-    if text or st.button('Analyze'):
-        predict(text)
+