Update app.py

app.py

@@ -38,13 +38,13 @@ def textclassification():
         st.write(f"Sentiment: {sentiment}")
         st.write(f"Confidence: {confidence:.2f}")
 
-        # Determine the rating based on confidence
+        # Determine the rating based on sentiment
         if sentiment == "POSITIVE":
-            # Scale confidence to a rating of 1 to 5
-            rating = int(confidence * 4) + 1 
-        else:
-            # For negative sentiment, invert the confidence score
-            rating = int((1 - confidence) * 4) + 1  
+            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")