Update app.py

app.py

@@ -36,7 +36,8 @@ if st.button('Load Model', disabled=False):
             return_tensors = 'pt'
             )
         device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        st.success("Model Loaded!")
+        with col1:
+            st.success("Model Loaded!")
         def predict(new_sentence):
             # We need Token IDs and Attention Mask for inference on the new sentence
             test_ids = []