Update app.py

app.py

@@ -11,7 +11,7 @@ tokenizer = BertTokenizer.from_pretrained(model_name, force_download=True)
 def inference(input_text):
     if "[MASK]" not in input_text:
         return "Error: The input text must contain the [MASK] token."
-    
+
     # Tokenisierung
     inputs = tokenizer(input_text, return_tensors="pt")
     mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
@@ -28,11 +28,21 @@ def inference(input_text):
     # Vorhersage in den Text einfügen
     predicted_token = tokenizer.decode(top_token)
     result_text = input_text.replace("[MASK]", predicted_token, 1)
-    
+
     return result_text
 
+
 # Gradio Interface definieren
-iface = gr.Interface()
+iface = gr.Interface(
     fn=inference,
+    inputs="text",
+    outputs="text",
+    examples=[
+        ["The capital of France is [MASK]."],
+        ["The quick brown fox jumps over the [MASK] dog."],
+    ],
+)
 
-    iface.launch(server_port=7862)
+# Interface starten
+if __name__ == "__main__":
+    iface.launch(server_port=7862)