prediction modfied

utils.py

@@ -63,9 +63,11 @@ def clean_text(text):
 
 
 def predict_cat(model, text):
-    p = int(model.predict(text,return_proba=True).max()*100)
-    cat =  model.predict(text)
-    return p,cat
+    
+    logits = model.predict(text,return_proba=True)
+    prob = int(logits.max()*100)
+    cat= label_df.iloc[logits.argmax()].values[0]
+    return prob,cat
 
 
 def grouper(iterable):