Update app.py

app.py

@@ -10,22 +10,34 @@ tokenizer = AutoTokenizer.from_pretrained("huimanho/CustomModel_Amazon")
 model = AutoModelForSequenceClassification.from_pretrained("huimanho/CustomModel_Amazon")
 
 # Streamlit app
-st.title("Chinese Review Analysis")
+st.title("Chinese Review Analysis - Translation, Rating & Sentiment")
 
 # Input text
-chinese_text = st.text_area("Enter Chinese Review:")
+chinese_text = st.text_area("Enter Chinese Review:", height=150)
 
 if st.button("Analyze"):
     # Translation
     english_text = pipe1(chinese_text)[0]['translation_text']
-    st.write("Translated Text:", english_text)
+    
+    # Display translation
+    st.subheader("Translated Text")
+    st.write(english_text)
 
     # Rating Prediction
     inputs = tokenizer(english_text, return_tensors="pt")
     outputs = model(**inputs)
     prediction = outputs.logits.argmax(-1).item()
-    st.write("Estimated Amazon Rating:", prediction + 1)
+
+    # Display estimated rating
+    st.subheader("Estimated Amazon Rating")
+    st.write(f"**Rating:** {prediction + 1} out of 5")     
 
     # Sentiment Classification
     sentiment = pipe3(english_text)[0]['label']
-    st.write("Sentiment:", sentiment)
+
+    # Display sentiment
+    st.subheader("Sentiment Analysis")
+    st.write(f"**Sentiment:** {sentiment}")    
+    
+    # Additional styling
+    st.markdown("---")  # Add a horizontal line for separation