Update app.py

app.py

@@ -29,11 +29,6 @@ def generation_model():
     generator = pipeline(model=model_name, tokenizer=model_name, task="text-generation")
     return generator
 
-@st.cache(show_spinner=False, allow_output_mutation=True)
-def sentiment_model():
-    sentiment_analysis = pipeline("sentiment-analysis")
-    return sentiment_analysis
-
 if option == "Extractive question answering":
     st.markdown("<h2 style='text-align: center; color:red;'>Extract answer from text</h2>", unsafe_allow_html=True)
     source = st.radio("How would you like to start? Choose an option below", ["I want to input some text", "I want to upload a file"])
@@ -103,15 +98,4 @@ elif option == "Text generation":
             generator = generation_model()
         with st.spinner(text="Generating text..."):
             generated_text = generator(text, max_length=50)
-            st.write(generated_text[0]["generated_text"])
-                
-elif option == "Sentiment analysis":
-    st.markdown("<h2 style='text-align: center; color:grey;'>Classify review</h2>", unsafe_allow_html=True)
-    text = st.text_input(label='Enter a sentence to get its sentiment analysis')
-    button = st.button("Get sentiment analysis")
-    if button:
-        with st.spinner(text="Loading sentiment analysis model..."):
-            sentiment_analysis = sentiment_model()
-        with st.spinner(text="Getting sentiment analysis..."):
-            sentiment = sentiment_analysis(text)
-            st.write(sentiment[0]["label"])
+            st.write(generated_text[0]["generated_text"])