Update app.py

app.py

@@ -70,7 +70,6 @@ with st.sidebar:
 ''')
 
 
-  
 st.subheader("YouTube Comments Sentiment Analysis", divider="red")
 tokenizer = transformers.DistilBertTokenizer.from_pretrained("tabularisai/robust-sentiment-analysis")
 model = transformers.DistilBertForSequenceClassification.from_pretrained("tabularisai/robust-sentiment-analysis")
@@ -178,8 +177,8 @@ if st.button("Sentiment Analysis", type="secondary"):
     else:
         st.warning(f"You have reached the maximum URL attempts ({max_attempts}).")
 
-st.write(f"URL pasted {st.session_state['url_count']} times.")
-            
+if 'url_count' in st.session_state: #added if statement.
+    st.write(f"URL pasted {st.session_state['url_count']} times.")