Update app.py

app.py

@@ -105,7 +105,7 @@ def infer_and_log(text_input):
             print(f"Error uploading log: {e}")
     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))
+    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():
@@ -118,7 +118,7 @@ with gr.Blocks() as app:
         input_box = gr.Textbox(label="Input Text", lines=10)
     with gr.Row():
         output_html = gr.HTML(label="Color-Coded Output")
-        output_json = gr.JSON(label="Word Probabilities")
+        output_json = gr.JSON(label="Word Probabilities",visible=False)
     with gr.Row():
         submit_btn = gr.Button("Submit")
         clear_btn = gr.Button("Clear")