Update app.py

app.py

@@ -3,21 +3,19 @@ from transformers import pipeline
 
 # Title and Subtitle
 st.title('Sentiment Analysis App')
-st.subheader('Enter text to analyze sentiment')
 
-# Text input box
-text = st.text_area('Enter your text')
+
+pipe=pipeline(model="dancingninjas/sentiment-model")
+text=st.text_area('Enter your text')
 
 if text:
-    # Analyze sentiment using the model
-    pipe = pipeline(model="dancingninjas/sentiment-model")
-    results = pipe(text)
+    out = pipe(text)
+
+    # Convert the model's output (0 or 1) to descriptive labels
+    sentiment_label = "Negative" if out[0]['label'] == 'LABEL_1' else "Positive"
 
-    # Determine sentiment label
-    sentiment_label = "Negative" if results[0]['label'] == 'LABEL_1' else "Positive"
+    # Display the sentiment label and probability
 
-    # Display sentiment label and probability
-    st.subheader('Sentiment Analysis Result:')
     st.write(f'Sentiment: {sentiment_label}')
-    st.write(f'Probability: {results[0]["score"]:.4f}')
+    st.write(f'Probability: {out[0]["score"]:.4f}')