add app.py

.gitignore

@@ -0,0 +1,3 @@
+.venv
+.vscode
+flagged

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+from transformers import pipeline
+
+unmasker = pipeline("fill-mask", model="anferico/bert-for-patents")
+
+
+def unmask(text):
+    res = unmasker(text)
+    out = {item["token_str"]: item["score"] for item in res}
+    return out
+
+
+textbox = gr.Textbox(label="Type patent abstract here", lines=5)
+
+demo = gr.Interface(
+    fn=unmask,
+    inputs=textbox,
+    outputs="label",
+    examples=[
+        [
+            "The present [MASK] provides a torque sensor that is small and highly rigid and for which high production efficiency is possible."
+        ],
+        [
+            "The present invention relates to [MASK] accessories and pertains particularly to a brake light unit for bicycles."
+        ],
+        [
+            "The present invention discloses a space-bound-free [MASK] and its coordinate determining circuit for determining a coordinate of a stylus pen."
+        ],
+        [
+            "The illuminated [MASK] includes a substantially translucent canopy supported by a plurality of ribs pivotally swingable towards and away from a shaft."
+        ],
+    ],
+)
+
+demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+torch
+transformers