Update app.py

app.py

@@ -1,21 +1,16 @@
 import streamlit as st
-from transformers import pipeline
-# Use a pipeline as a high-level helper
-from transformers import pipeline
-
-pipe = pipeline("text-classification", model="OatNapat/finetuned_yelp")
-# Load model directly
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
+# Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("OatNapat/finetuned_yelp")
 model = AutoModelForSequenceClassification.from_pretrained("OatNapat/finetuned_yelp")
 
+# Create a sentiment analysis pipeline
 nlp = pipeline("sentiment-analysis", model=model)
 
-
 st.title("Sentiment Analysis App")
 user_input = st.text_input("Enter a sentence for sentiment analysis:")
 if user_input:
     result = nlp(user_input)
     sentiment = result[0]["label"]
-    st.write(f"Sentiment: {sentiment}")
+    st.write(f"Sentiment: {sentiment}")