Update app.py

app.py

@@ -7,7 +7,13 @@ API_TOKEN = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
 HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
 def get_sentiment_category(sentiment_label):
-    return sentiment_label.lower()  # Convert the label to lowercase
+    sentiment_label = sentiment_label.lower()
+    if "positive" in sentiment_label:
+        return "Positive"
+    elif "negative" in sentiment_label:
+        return "Negative"
+    else:
+        return "Mixed"
 
 st.title("DistilGPT2 Movie Review Sentiment Analysis")
 
@@ -39,9 +45,9 @@ if st.button("Analyze"):
         sentiment_end = result.find(".", sentiment_start)
         sentiment_label = result[sentiment_start:sentiment_end].strip()
 
-        # Convert sentiment label to lowercase
+        # Convert sentiment label to category
         sentiment_category = get_sentiment_category(sentiment_label)
         st.write(f"Sentiment: {sentiment_category}")
 
     except requests.exceptions.RequestException as e:
-        st.error("Error reaching API\n{}".format(e))
+        st.error("Error reaching API\n{}".format(e))