Update app.py

app.py

@@ -10,6 +10,10 @@ model = AutoModelForMaskedLM.from_pretrained(model_name)
 nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
 def predict(text):
+    # Check if the input text contains the [MASK] token
+    if "[MASK]" not in text:
+        return "Error: Please enter a sentence containing a [MASK] token."
+    
     # Process the text using the model
     result = nlp(text)
     return result
@@ -23,4 +27,4 @@ iface = gr.Interface(
     description="Test BlueBERT on biomedical data or general text"
 )
 
-iface.launch(share=True)
+iface.launch()