Update app.py

app.py

@@ -122,12 +122,8 @@ def infer_and_log(text_input):
             print(f"Uploaded log {submission_id}")
         except Exception as e:
             print(f"Error uploading log: {e}")
-    # tokens = text_input.split()
-    # Replace \n with <br> in the input text for HTML rendering
-    tokens = text_input.replace("\n", " \n ").split()
-    
-    formatted_output = " ".join(f'<span style="color:{color}">{token}</span>' if token != "\n" else "<br>" for token, color in zip(tokens, word_colors))
-    # formatted_output = " ".join(f'<span style= "color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
+    tokens = text_input.split()
+    formatted_output = " ".join(f'<span style= "color:{color}">{token}</span>' for token, color in zip(tokens, word_colors))
     return formatted_output, word_probs
 
 def clear_fields():