Update app.py

app.py

@@ -4,21 +4,26 @@ from transformers import pipeline
 # Load the fill-mask pipeline with the specified model
 @st.cache_resource
 def load_pipeline():
+    """Load the ParlBERT model for the fill-mask task."""
     pipe = pipeline("fill-mask", model="InfAI/parlbert-german-law")
     return pipe
 
+# Load the pipeline
 pipe = load_pipeline()
 
 # Streamlit app UI
 st.title("ParlBERT German Law Fill-in-the-Blank Assistant")
 st.write("Enter a sentence with a `[MASK]` token, and the model will predict possible words.")
 
-# Input text area for the sentence
+# Input text area for the sentence with a default example
 user_input = st.text_input("Your sentence (use [MASK] as a placeholder):", "Das Gesetz [MASK] den Bürger.")
 
-# Generate predictions
+# Generate predictions when the button is clicked
 if st.button("Generate Prediction"):
+    # Run the model on the input sentence
     predictions = pipe(user_input)
+    
+    # Display predictions
     st.write("Predicted Fill-ins:")
     for i, pred in enumerate(predictions):
-        st.write(f"{i+1}. {pred['sequence']} (Score: {pred['score']:.4f})")
+        st.write(f"{i+1}. {pred['sequence']} (Score: {pred['score']:.4f})")