NIXBLACK commited on
Commit
6068c42
·
1 Parent(s): 3277e5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -32,8 +32,15 @@ label_encoder.fit(target_classes)
32
  user_text_embedding = encoder.encode_sentences([user_text])[0]
33
  user_text_embedding = np.reshape(user_text_embedding, (1, -1))
34
 
35
- predicted_sentiment = np.argmax(model.predict(user_text_embedding))
36
- predicted_sentiment_no = label_encoder.inverse_transform([predicted_sentiment])[0]
 
 
 
 
 
 
 
37
 
38
  if predicted_sentiment_no == 1:
39
  predicted_sentiment_label = 'neutral'
@@ -42,4 +49,4 @@ elif predicted_sentiment_no == 2:
42
  else:
43
  predicted_sentiment_label = 'negative'
44
 
45
- st.write("Predicted Sentiment:" + predicted_sentiment_label)
 
32
  user_text_embedding = encoder.encode_sentences([user_text])[0]
33
  user_text_embedding = np.reshape(user_text_embedding, (1, -1))
34
 
35
+ st.write("Debug: user_text_embedding", user_text_embedding)
36
+
37
+ try:
38
+ predicted_sentiment = np.argmax(model.predict(user_text_embedding))
39
+ predicted_sentiment_no = label_encoder.inverse_transform([predicted_sentiment])[0]
40
+ st.write("Debug: predicted_sentiment_no", predicted_sentiment_no)
41
+ except Exception as e:
42
+ st.write("Error during prediction:", str(e))
43
+ raise e
44
 
45
  if predicted_sentiment_no == 1:
46
  predicted_sentiment_label = 'neutral'
 
49
  else:
50
  predicted_sentiment_label = 'negative'
51
 
52
+ st.write("Predicted Sentiment:", predicted_sentiment_label)