Update app.py

app.py

@@ -5,6 +5,7 @@ from transformers import AutoTokenizer, AutoModelWithLMHead
 from transformers import pipeline
 
 #tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
+sentiment_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-imdb-sentiment")
 
 @st.cache
 def load_model(model_name):
@@ -17,9 +18,8 @@ def load_text_gen_model():
     
 @st.cache
 def get_sentiment_model():
-    sentiment_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-imdb-sentiment")
     sentiment_model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-imdb-sentiment")
-    return sentiment_model ,sentiment_tokenizer 
+    return sentiment_model 
     
 def get_sentiment(text):
     input_ids = sentiment_tokenizer .encode(text + '</s>', return_tensors='pt')
@@ -28,7 +28,7 @@ def get_sentiment(text):
     label = dec[0]
     return label
   
-sentiment_model ,sentiment_tokenizer  = get_sentiment_model()
+sentiment_model   = get_sentiment_model()
 text_generator = load_text_gen_model()
 
 action = st.sidebar.selectbox("Pick an Action", ["Analyse a Review","Generate an Article","Create an Image"])