Update app.py

app.py

@@ -38,20 +38,12 @@ def textclassification():
         st.write(f"Sentiment: {sentiment}")
         st.write(f"Confidence: {confidence:.2f}")
 
-        # Determine the rating based on sentiment
-        if sentiment == "POSITIVE":
-            rating = 5  # Assign a rating of 5 stars for positive sentiment
-        elif sentiment == "NEUTRAL":
-            rating = 3  # Assign a neutral rating of 3 stars
-        else:  # Assuming the other sentiment is "NEGATIVE"
-            rating = 1  # Assign a rating of 1 star for negative sentiment
-
         # Display the rating
-        st.write(f"The rating is {rating} stars")
+        st.write(f"The rating is {sentiment} stars")
 
         # Question Answering
         qa_input = {
-            'question': f'Why is the rating {rating} star?',
+            'question': f'Why is the rating {sentiment} star?',
             'context': user_input  # Use user input as context
         }
         qa_result = qa_pipeline(qa_input)