Update app.py

app.py

@@ -90,6 +90,11 @@ def process_api(text):
     # Seq_Predicted = Seq_model.predict(padded_sequence)
     # predicted_label_index = np.argmax(Seq_Predicted)
 
+    # ----------- Proba -----------
+    Logistic_Predicted_proba = logistic_model.predict_proba(processed_text)
+    svm_new_probs = SVM_model.decision_function(processed_text)
+    svm_probs = svm_model.predict_proba(svm_new_probs)
+    
     # ----------- Debug Logs -----------
     logistic_debug = decodedLabel(int(Logistic_Predicted[0]))
     svc_debug = decodedLabel(int(SVM_Predicted[0]))
@@ -98,8 +103,14 @@ def process_api(text):
     print('SVM', int(SVM_Predicted[0]), svc_debug)
     
     return {
-            'Logistic_Predicted':decodedLabel(int(Logistic_Predicted[0])),
-            'SVM_Predicted': decodedLabel(int(SVM_Predicted[0])),
+            'Logistic_Predicted':{
+                'predicted_label': decodedLabel(int(Logistic_Predicted[0])),
+                'probability': int(float(np.max(Logistic_Predicted_proba))*10000//100)
+            },
+            'SVM_Predicted': {
+                'predicted_label': decodedLabel(int(SVM_Predicted[0])),
+                'probability': int(float(np.max(svm_probs))*10000//100)
+            },
             # 'LSTM': decodedLabel(int(predicted_label_index)),
             'Article_Content': text
         }