Update app.py

app.py

@@ -9,13 +9,8 @@ from transformers import BertForSequenceClassification, AdamW, BertConfig
 st.set_page_config(layout='wide', initial_sidebar_state='expanded')
 st.title("Spamd: Turkish Spam Detector")
 st.markdown("Enter the text you'd like to analyze for spam.")
-text = st.text_input("Enter the text you'd like to analyze for spam.")
+text = st.text_input()
 
-"""Spamd_SpamDetector_Turkish_BERT_22.09.2022.ipynb
-
-Original file is located at
-    https://colab.research.google.com/drive/1QuorqAuLsmomesZHsaQHEZgzbPEM8YTH
-"""
 
 
 
@@ -70,25 +65,9 @@ def predict(new_sentence):
       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'
-    st.write('Predicted Class: ', prediction)
+    st.header('Predicted Class: ', prediction)
     
     #st.write('Input', namestr(new_sentence, globals()),': \n', new_sentence) 
       # Remove the namestr(new_sentence, globals()) in case of an error      
 if st.button('Analyze'):
     predict(text)   
-
-
-
-
-'''
-@software{stefan_schweter_2020_3770924,
-  author       = {Stefan Schweter},
-  title        = {BERTurk - BERT models for Turkish},
-  month        = apr,
-  year         = 2020,
-  publisher    = {Zenodo},
-  version      = {1.0.0},
-  doi          = {10.5281/zenodo.3770924},
-  url          = {https://doi.org/10.5281/zenodo.3770924}
-}
-'''