chore: add bert raw output.

app.py

@@ -262,9 +262,12 @@ def generate_answer(text):
 
     # Format Prediction
     label, score = extract_max_label_score(predictions[0])
+    entities_formatted["NIVELES_CASA"] = label
+    entities_formatted["prob_NIVELES_CASA"] = score
     prediction_cleaned = clean_prediction_bert(label, score)
     if isinstance(prediction_cleaned, int):
         entities_cleaned["NIVELES_CASA"] = prediction_cleaned
+    
 
     result_json = json.dumps(entities_cleaned, indent = 4, ensure_ascii = False)