Update app.py

app.py

@@ -5,12 +5,9 @@ import streamlit as st
 import tweepy
 from plotly.subplots import make_subplots
 from transformers import pipeline
-consumer_key = "kG4NXwrJllh7Jv5aLA9yjfb1U"
-consumer_secret = "fH27zr7ZcqYdbQMOSPY3v5a6nEgcOXDyFCJPFSb0VNNinZafCz"
-access_key = "1116912581434695680-SA7ddRFq6GUxISNrL1V5IoN2Z9FK3m"
-access_secret = "JDu1Rj4tj8kSilqawlH88LU8Y7nyu9GcbNZygNCpTk9kd"
-auth = tweepy.OAuthHandler(consumer_key,consumer_secret)
-auth.set_access_token(access_key,access_secret)
+
+auth = tweepy.OAuthHandler(st.secrets["consumer_key"], st.secrets["consumer_secret"])
+auth.set_access_token(st.secrets["access_key"], st.secrets["access_secret"])
 api = tweepy.API(auth)
 
 
@@ -60,14 +57,19 @@ def get_aggregation_period(df):
         return "30D"
 
 
-@st.cache_data
+@st.cache(allow_output_mutation=True)
 def load_model():
     pipe = pipeline(task="sentiment-analysis", model="bhadresh-savani/distilbert-base-uncased-emotion")
     return pipe
 
 
+"""
+# Twitter Emotion Analyser
+"""
+
+
 pipe = load_model()
-twitter_handle = st.sidebar.text_input("Twitter handle:", "elonmusk")
+twitter_handle = st.sidebar.text_input("Twitter handle:", "huggingface")
 twitter_count = st.sidebar.selectbox("Number of tweets:", (10, 100, 500, 1000, 3200))
 
 
@@ -130,4 +132,4 @@ if st.sidebar.button("Get tweets!"):
     st.plotly_chart(fig)
 
     # tweet sample
-    st.markdown(df.sample(n=5).to_markdown())
+    st.markdown(df.sample(n=5).to_markdown())