DeeeTeeee01 commited on
Commit
9dbd96b
·
1 Parent(s): 1f9eb21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -171,7 +171,13 @@ text = st.text_input("Enter text:")
171
  predict_button = st.button("Predict")
172
 
173
  # Prediction output
174
- if predict_button and text:
 
 
 
 
 
 
175
 
176
  # Get probabilities
177
  negative, positive, neutral = predict_sentiment(text)
 
171
  predict_button = st.button("Predict")
172
 
173
  # Prediction output
174
+ def predict_sentiment(text):
175
+
176
+ inputs = tokenizer(text, return_tensors="pt")
177
+
178
+ outputs = model(**inputs)
179
+
180
+ # Rest of code to extract probabilities
181
 
182
  # Get probabilities
183
  negative, positive, neutral = predict_sentiment(text)