Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from transformers import pipeline
+fill_mask = pipeline("fill-mask", model="bert-base-uncased")
+def complete_sentence(text):
+    if "[MASK]" not in text:
+        return "Please include [MASK] token in your input text"
+    results = fill_mask(text)
+    output = []
+    for result in results:
+        completed = f"Token: {result['token_str']} (Score: {result['score']:.3f})"
+        output.append(completed)
+    return "\n".join(output)
+iface = gr.Interface(
+    fn=complete_sentence,
+    inputs=gr.Textbox(
+        lines=3, 
+        placeholder="Enter text with [MASK] token...",
+        label="Input Text"
+    ),
+    outputs=gr.Textbox(
+        lines=5,
+        label="Completed Sentences"
+    ),
+    title="Sentence Completion",
+    description="Enter a sentence with [MASK] token and get predictions for the masked word",
+    examples=[
+        ["The [MASK] is bright today."],
+        ["I love to [MASK] in my free time."],
+        ["She [MASK] to the store yesterday."]
+    ],
+    theme=gr.themes.Soft()
+)
+iface.launch()
+
+if __name__ == '__main__':
+    demo.launch()