Update app.py

app.py

@@ -1,13 +1,14 @@
+import streamlit as st
+from transformers import pipeline
+sentiment_task = pipeline("sentiment-analysis", model=model_path, tokenizer=model_path)
 
-import torch
-from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
+st.title("Sentiment Analysis App")
+userinput=st.text_area("Enter text for analysis")
 
-tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
-model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
+if userinput:
+    result=sentiment_task(userinput)
 
-inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
-with torch.no_grad():
-    logits = model(**inputs).logits
+    st.write("Sentiment analysis result:")
+    st.write(f"label]:{ result[0]['label']}]")
+    st.write(f"confidence]:{ result[0]['score']:,4f}]")
 
-predicted_class_id = logits.argmax().item()
-model.config.id2label[predicted_class_id]