Update app.py

app.py

@@ -30,10 +30,14 @@ if st.button("Analyze"):
         response = requests.post(API_URL, headers=HEADERS, json={"inputs": prompt}, timeout=10)
         response.raise_for_status()
 
-        result_score = float(response.json()[0]['generated_text'].split('/')[0])
-        sentiment = "positive" if result_score > 5.0 else "negative" if result_score < 5.0 else "mixed"
-        
-        st.write(f"Sentiment: {sentiment}")
+        # Extract sentiment label from the result
+        result = response.json()[0]['generated_text']
+        sentiment = result.split()[-1].lower()
+
+        if sentiment in ("positive", "negative", "mixed"):
+            st.write(f"Sentiment: {sentiment}")
+        else:
+            st.write("Could not determine sentiment")
 
     except requests.exceptions.Timeout:
         st.error("Request to the API timed out. Please try again.")