Initial version

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from transformers import pipeline
+
+# token_classifier = pipeline(
+#   model="ZurichNLP/swissbert-ner",
+#   aggregation_strategy="simple",
+# )
+
+def run_ner_placeholder(text: str, language: str):
+    assert language in {"de_CH", "fr_CH", "it_CH", "rm_CH"}, f"Language {language} not supported"
+    # Placeholder
+    assert text == "Mein Name sei Gantenbein."
+    return [{'entity_group': 'PER',
+      'score': 0.50026333,
+      'word': 'Gantenbein',
+      'start': 13,
+      'end': 24}
+    ]
+
+def run_ner(text: str, language: str):
+  token_classifier.model.set_default_language(language)
+  output = token_classifier(text)
+  return output
+
+def visualize_ner(text: str, language: str):
+  labels = run_ner_placeholder(text, language)
+  return {
+    "entities": labels,
+    "text": text,
+  }
+
+demo = gr.Interface(
+    fn=visualize_ner,
+    inputs=[
+        gr.Textbox(lines=5, placeholder="Enter your text here...", label="Text"),
+        gr.Dropdown(choices=[
+            ("German", "de_CH"),
+            ("French", "fr_CH"),
+            ("Italian", "it_CH"),
+            ("Romansh", "rm_CH")
+        ], value="de_CH", label="Language")
+    ],
+    outputs=gr.HighlightedText(
+      label="Named Entities",
+      combine_adjacent=True,
+    ),
+    title="SwissBERT NER",
+    description="Analyze text with SwissBERT for named entity recognition.",
+    allow_flagging=False,
+    examples=[
+        ["Es war die grosse Überraschung im Baselbiet bei den gestrigen Wahlen: Thomi Jourdan von der Kleinstpartei EVP hat den Sprung in die Regierung geschafft und sich gegen SVP-Nationalrätin Sandra Sollberger durchgesetzt.", "de_CH"],
+        ["La cheffe de la Direction suisse du développement et de la coopération (DDC), Patricia Danzi, a tenu à accueillir elle-même les 87 spécialistes à l'aéroport, a indiqué le Département fédéral des affaires étrangères (DFAE). Elle était notamment accompagnée de Silvio Flückiger, chef suppléant de l'Aide humanitaire de la Suisse, ainsi que du chef de l'armée Thomas Süssli.", "fr_CH"],
+        ["I Grigioni, al contrario di quanto fatto dal Ticino, non intendono adottare una legislazione particolare per fronteggiare la diffusione delle sigarette elettroniche tra i ragazzi. Lo ha ribadito martedì il Consiglio di Stato rispondendo alle preoccupazioni espresse dall'ex direttrice scolastica e deputata del Centro di Mesocco Piera Furger. Di fronte al Gran Consiglio, il Governo ha spiegato di privilegiare una soluzione a livello nazionale e un approccio all'insegna della prevenzione e non della repressione.", "it_CH"],
+        ["Tar l’Ospital chantunal dal Grischun datti in augment d’accidents da skiunzas e snowboardists sin pista da 20%. Quai dal prim da schaner fin ils 6 da favrer, cumpareglià cun la media dals tschintg onns passads. Quai rapporta il «Tagesanzeiger» Tenor il pledader da medias, Dajan Roman, haja era dà dapli blessuras grevas. Ils ultims onns era la media tar 3%, dapi l’entschatta da l’onn èn quai gia 4,2%.", "rm_CH"],
+    ],
+)
+
+demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+transformers