use ner and rel

app.py

@@ -25,7 +25,6 @@ relations = {
     ]
 }
 
-# The sample sentence you want to highlight:
 SAMPLE_TEXT = (
     "The Jordan Home Visits Survey, Round II (HV), was carried out by UNHCR and the World Food "
     "Programme between November 2013 and September 2014. Through in-home visits to Syrian refugee "
@@ -42,7 +41,7 @@ def highlight_text(text):
             "start":  ent["start"],
             "end":    ent["end"],
         })
-    # 2) RE spans: annotate each target with its relation label
+    # 2) RE spans
     for src, rels in relations.items():
         for r in rels:
             label = r["relation"]
@@ -57,16 +56,15 @@ def highlight_text(text):
 
 with gr.Blocks() as demo:
     gr.Markdown("## Data Use Detector\n"
-                "Input text and the model will highlight the entities it detects.")
-    
+                "Click **Highlight** to run inference on the sample sentence.")
+
     txt_in  = gr.Textbox(label="Input Text", lines=4, value=SAMPLE_TEXT)
-    btn     = gr.Button("Highlight Entities")
+    btn     = gr.Button("Highlight")
     txt_out = gr.HighlightedText(label="Annotated Entities")
-    
+
+    # **Only** trigger on button click
     btn.click(fn=highlight_text, inputs=txt_in, outputs=txt_out)
-    txt_in.submit(fn=highlight_text, inputs=txt_in, outputs=txt_out)
-    demo.load(fn=highlight_text, inputs=txt_in, outputs=txt_out)
-    
+
     gr.Markdown("""
 **Legend**  
 - **named dataset** → Home Visits Survey