"~"

app.py

@@ -7,6 +7,8 @@ from model import SeqTagger
 from dataset import SeqTaggingClsDataset
 from typing import Dict
 import torch
+from gradio.components import Textbox
+import random
 
 # Disable cudnn to ensure the model runs on CPU
 torch.backends.cudnn.enabled = False
@@ -65,7 +67,7 @@ best_model.load_state_dict(checkpoint['model_state_dict'])
 # Set the model to evaluation mode
 best_model.eval()
 
-def classify(text: str):
+def tagging(text: str):
     # Tokenize the text
     str_text = [str(text.split())]
     dic_text = {"tokens": str_text, "tags": [None], "id": ["text-0"]}
@@ -87,33 +89,51 @@ def classify(text: str):
 
     text_tags = []
     for i, tag in enumerate(preds[0]):
-        if tag == "O":
-            text_tags.extend([(text.split()[i], None), (" ", None)])
-        else:
-            text_tags.extend([(text.split()[i], tag), (" ", None)])
-
+        text_tags.extend([(text.split()[i], tag), (" ", None)])
     return text_tags
 
-# Create a Gradio interface
-demo = gr.Interface(
-    classify,
-    gr.Textbox(placeholder="Please enter a text..."),
-    gr.HighlightedText(),
-    interpretation="none",
-    live=False,
-    enable_queue=True,
-    examples=[
-        ["i have three people for august seventh"],
-        ["a table for 2 adults and 4 children please"],
-        ["i have a booking tomorrow for chara conelly at 9pm"],
-        ["me and 4 others will be there at 8:30pm"],
-        ["probably malik belliard has done the booking and it is on in 10 days"],
-        ["i want to book a table for me and my wife tonight at 6 p.m"],
-        ["date 18th of december"]
-    ],
-    title="Slot Tagging",
-    description="This is a demo for slot tagging. Enter a sentence, and it will predict and highlight the slots."
-)
-
-# Launch the Gradio interface
-demo.launch(share=True)
+examples=[
+        "i have three people for august seventh",
+        "a table for 2 adults and 4 children please",
+        "i have a booking tomorrow for chara conelly at 9pm",
+        "me and 4 others will be there at 8:30pm",
+        "probably malik belliard has done the booking and it is on in 10 days",
+        "i want to book a table for me and my wife tonight at 6 p.m",
+        "date 18th of december",
+        "The concert is on September fifteenth",
+        "I need a reservation for a party of eight on Sunday",
+        "Her birthday is on May twenty-third",
+        "We have a meeting at ten a.m. tomorrow",
+        "The conference starts at eight o'clock in the morning",
+        "He booked a flight for February seventh",
+        "There is an event on the twenty-ninth of June",
+        "Please reserve a table for two for this evening",
+        "The project deadline is on March fourth",
+        "We'll have a gathering on the first of July"
+]
+
+def random_sample():
+        random_number = random.randint(0, len(examples) - 1)
+        return examples[random_number]
+
+description="""
+# Slot Tagging
+This is a demo for slot tagging. Enter a sentence, and it will predict and highlight the slots.
+"""
+
+title="Slot Tagging"
+
+with gr.Blocks(theme=gr.themes.Soft(), title=title) as demo:
+        gr.Markdown(description)
+
+        with gr.Row():
+                C_input = Textbox(lines=3, label="Context", placeholder="Please enter a text...")
+                T_output = gr.HighlightedText(lines=3, label="IOB Tagging")
+        with gr.Row():
+                random_button = gr.Button("Random")
+                tagging_button = gr.Button("Tagging")
+
+        random_button.click(random_sample, inputs=None, outputs=C_input)
+        tagging_button.click(tagging, inputs=C_input, outputs=T_output)
+
+demo.launch()