Update app.py

app.py

@@ -5,16 +5,6 @@ import torch
 # Set up the device (GPU or CPU)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-# Function to perform sentiment analysis
-def perform_sentiment_analysis(text):
-    inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
-    inputs = inputs.to(device)
-    outputs = model(**inputs)
-    logits = outputs.logits
-    probabilities = torch.softmax(logits, dim=1).detach().cpu().numpy()[0]
-    sentiment_label = "Positive" if probabilities[1] > probabilities[0] else "Negative"
-    return sentiment_label, probabilities
-
 # Streamlit app
 def main():
     st.title("Sentiment Analysis App")
@@ -36,7 +26,13 @@ def main():
     tokenizer = AutoTokenizer.from_pretrained(model_name)
 
     if st.button("Analyze"):
-        sentiment_label, probabilities = perform_sentiment_analysis(text)
+        # Perform sentiment analysis
+        inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
+        inputs = inputs.to(device)
+        outputs = model(**inputs)
+        logits = outputs.logits
+        probabilities = torch.softmax(logits, dim=1).detach().cpu().numpy()[0]
+        sentiment_label = "Positive" if probabilities[1] > probabilities[0] else "Negative"
         st.write(f"Sentiment: {sentiment_label}")
         st.write(f"Positive probability: {probabilities[1]}")
         st.write(f"Negative probability: {probabilities[0]}")