change css

app.py

@@ -16,27 +16,17 @@ def ner(text, file_names):
     result = run(tokenizer, model, gazetteers_for_matching, text, file_names)
     return {"text": text, "entities": result}
 
-with gr.Blocks(css="footer{display:none !important}", theme=gr.themes.Default(primary_hue="blue", secondary_hue="sky")) as demo:
-# with gr.Blocks(theme=gr.themes.Soft()) as demo:
-
+with gr.Blocks(css="./style.css", theme=gr.themes.Default(primary_hue="blue", secondary_hue="sky")) as demo:
     gr.Interface(ner,
         gr.Textbox(lines=10, placeholder="Enter sentence here..."), 
-        gr.HighlightedText(show_legend=True, color_map={"PER": "red", "ORG": "green", "LOC": "blue"}),
+        # gr.HighlightedText(show_legend=True, color_map={"PER": "red", "ORG": "green", "LOC": "blue"}),
+        gr.HighlightedText(show_legend=True, color_map={"PER": "#660606", "ORG": "#066618", "LOC": "#061966"}, elem_id="highlighted_text"),
         examples=examples,
         title="NerROB-czech",
         description="This is an implementation of a Named Entity Recognition model for the Czech language using gazetteers.",        
         allow_flagging="never",
-        additional_inputs=gr.File(label="Upload a JSON file", file_count="multiple", file_types=[".json"]),
-        additional_inputs_accordion=gr.Accordion(label="Additional Gazetters", open=False)
+        additional_inputs=gr.File(label="Upload a JSON file containing gazetteers", file_count="multiple", file_types=[".json"]),
         )
 
-    # gr.Interface(load_gazetters,
-    #     gr.File(label="Upload a JSON file", file_count="multiple", file_types=[".json"]),
-    #     None,
-    #     allow_flagging="never",
-    #     description="Here you can upload your own gazetteers.", )
-    # btn = gr.Button(value="Change Language")
-    # btn.click(reload)
-
 if __name__ == "__main__":
     demo.launch()

style.css

@@ -0,0 +1,22 @@
+footer {
+    display: none !important;
+}
+
+.container .svelte-ju12zg {
+    color: white;
+}
+
+.text.svelte-ju12zg {
+    padding: 0;
+    margin: 0;
+}
+
+.textspan.svelte-ju12zg.hl {
+    margin: 0;
+    padding: 0;
+}
+
+.textspan.svelte-ju12zg.no-cat {
+    margin: 0;
+    padding: 0;
+}

website_script.py

@@ -51,8 +51,6 @@ def run(tokenizer, model, gazetteers, text, file_names=None):
     tokenized_inputs = tokenizer(
         text, truncation=True, is_split_into_words=False, return_offsets_mapping=True
     )
-    for i in gazetteers_for_matching:
-        print(len(i))
     matches = gazetteer_matching(text, gazetteers_for_matching)
     new_g = []
     word_ids = tokenized_inputs.word_ids()