Update app.py

app.py

@@ -1,7 +1,8 @@
 # 🔍 Masked Word Predictor | CPU-only HF Space
 
 import gradio as gr
-from transformers import pipeline, PipelineException
+from transformers import pipeline
+from transformers.pipelines.base import PipelineException  # correct import
 
 # Load the fill-mask pipeline once at startup
 fill_mask = pipeline("fill-mask", model="distilroberta-base", device=-1)
@@ -16,9 +17,9 @@ def predict_mask(sentence: str, top_k: int):
 
     # If no mask token present, show error
     if mask not in sentence:
-        return [{"sequence": f"Error: please include `[MASK]` in your sentence.", "score": 0.0}]
+        return [{"sequence": "Error: please include `[MASK]` in your sentence.", "score": 0.0}]
 
-    # Call the pipeline and catch any unexpected exceptions
+    # Call the pipeline and catch any pipeline-specific exceptions
     try:
         preds = fill_mask(sentence, top_k=top_k)
     except PipelineException as e: