Update app.py

app.py

@@ -146,6 +146,10 @@ tokenizer = transformers.AutoTokenizer.from_pretrained("DeeeTeeee01/mytest_train
 
 @st.cache_resource
 def predict_sentiment(text):
+
+  inputs = tokenizer(text, return_tensors="pt")
+
+  outputs = model(**inputs)
   
   # Get full model outputs 
   outputs = model(text)
@@ -171,14 +175,8 @@ text = st.text_input("Enter text:")
 predict_button = st.button("Predict") 
 
 # Prediction output
-def predict_sentiment(text):
-
-  inputs = tokenizer(text, return_tensors="pt")
-
-  outputs = model(**inputs)
+if predict_button and text:
   
-  # Rest of code to extract probabilities
-
   # Get probabilities
   negative, positive, neutral = predict_sentiment(text)