Update app.py

app.py

@@ -11,7 +11,10 @@ nlp_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
 # Function to interact with ClinicalBERT
 def medical_chatbot(user_input):
-    response = nlp_pipeline(user_input.replace("[MASK]", ""))
+    if "[MASK]" not in user_input:
+        return "Please include '[MASK]' in your input. Example: 'The patient is showing signs of [MASK].'"
+
+    response = nlp_pipeline(user_input)
     return response[0]["sequence"]  # Returns the most likely sentence
 
 # Gradio UI