Update app.py

app.py

@@ -10,7 +10,7 @@ def summarization(text):
 # Sentiment Classification
 def sentiment_classification(summary):
     sentiment_model = pipeline("text-classification", model="wxrrrrrrr/finetunde_sentiment_analysis")
-    result = sentiment_model(summary, max_length=100, truncation=True)[0]
+    result = sentiment_model(summary, max_length=100, truncation=True)[0]['label']
     return result
 
 def main():
@@ -26,13 +26,12 @@ def main():
 
         # Stage 2: Sentiment Classification
         st.text('Analyzing sentiment...')
-        result = sentiment_classification(summary)
-            sentiment = result['label']
-            score = result['score']
-            st.write("Sentiment:", sentiment)
-            st.write("Score:", score)
+        sentiment = sentiment_classification(summary)
+        st.write(sentiment)
+
         # Display the classification result
         st.write("Sentiment:", sentiment)
 
+    
 if __name__ == '__main__':
     main()