Update app.py

app.py

@@ -37,7 +37,6 @@ stop_english = set(stopwords.words("english"))
 # -----------------------------
 st.write("Account Disruption")
 st.write("""Dear Customer Support Team,
-
 I am writing to report a significant problem with the centralized account management portal...
 """)
 
@@ -58,7 +57,15 @@ with col2:
 model_path = "model.h5"
 model = load_model(model_path, compile=False)  # <- works on HF
 
+with open("le_type.pkl", "rb") as f:
+    le_type = pickle.load(f)
+
+with open("le_queue.pkl", "rb") as f:
+    le_queue = pickle.load(f)
 
+with open("mlb.pkl", "rb") as f:
+    mlb = pickle.load(f)
+    
 # -----------------------------
 # Load Tokenizer
 # -----------------------------
@@ -114,3 +121,10 @@ if st.button("Submit"):
 
     preds = model.predict(seq)
     st.write("Model Output:", preds)
+    pred_type_probs, pred_queue_probs, pred_tags_probs = preds
+    pred_type_labels = le_type.inverse_transform(np.argmax(pred_type_probs, axis=1))
+    pred_queue_labels = le_queue.inverse_transform(np.argmax(pred_queue_probs, axis=1))
+    pred_tags_binary = (pred_tags_probs >= 0.5).astype(int)
+    pred_tags_lists = mlb.inverse_transform(pred_tags_binary)
+
+    st.write(pred_tags_lists)