Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import pipeline
 summarizer = pipeline("summarization", model="Falconsai/text_summarization")
 
 # Load the text classification model pipeline
-classifier = pipeline("text-classification", model='Rainess/Finturned-Music-Sentiment', return_all_scores=True)
+classifier = pipeline("text-classification", model='Rainess/Finturned-Music-Sentiment')
 
 def summary_song(text):
     input_length = len(text)
@@ -38,10 +38,7 @@ if st.button("Summarize and Classify"):
         # Perform text classification on the summarized text
         classification_results = classifier(summarized_text)
         
-        # Get the highest scoring sentiment
-        max_score = float('-inf')
-        max_label = ''
-        
+
         
         # Map label to sentiment
         sentiment = "Positive" if max_label == "1" else "Negative"
@@ -50,7 +47,7 @@ if st.button("Summarize and Classify"):
         st.write("Original Lyrics:", text)
         st.write("Summarized Text:", summarized_text)
         st.write("Sentiment:", sentiment)
-        st.write("Score:", max_score)
+
     else:
         st.write("Please enter some lyrics to summarize and classify.")