Update app.py

app.py

@@ -106,7 +106,7 @@ def infer_and_log(text_input):
     tokens = text_input.split()
     
     formatted_output = " ".join(f'<span style="background-color:#1F1F23; color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
-    return formatted_output
+    return formatted_output, word_probs
 
 def clear_fields():
     return "", ""