Update app.py

app.py

@@ -12,13 +12,13 @@ def predict(text):
     
     # Splitting the input text into words
     words = text.split()
-    # Highlighting the third word if it exists
+    # Highlighting the third word if it exists with a background color
     if len(words) >= 3:
-        words[2] = f"<strong>{words[2]}</strong>"
+        words[2] = f"<span style='background-color: yellow;'>{words[2]}</span>"
     highlighted_text = " ".join(words)
     
     # Adding the highlighted input text at the beginning of the response, followed by a new line
-    response = f"Input: {highlighted_text}\n\n"  # Modified to include highlighted third word
+    response = f"Input: {highlighted_text}\n\n"
     
     results = text_pipe(text)
     all_scores = results[0]