Update app.py

app.py

@@ -61,7 +61,7 @@ if st.button('Load Model'):
         
             # Forward pass, calculate logit predictions
             with torch.no_grad():
-              output = model(test_ids.to(device), token_type_ids = None, attention_mask = test_attention_mask.to(device))
+                output = model(test_ids.to(device), token_type_ids = None, attention_mask = test_attention_mask.to(device))
         
             prediction = 'Spam' if np.argmax(output.logits.cpu().numpy()).flatten().item() == 1 else 'Normal'
             pred = 'Predicted Class: '+ prediction
@@ -70,7 +70,7 @@ if st.button('Load Model'):
             
             #st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence) 
            with col2:
-            text = st.text_input("Enter the text you'd like to analyze for spam.")
-            if text or st.button('Analyze'):
-                predict(text)
+               text = st.text_input("Enter the text you'd like to analyze for spam.")
+               if text or st.button('Analyze'):
+                   predict(text)
     st.success()