Update app.py

app.py

@@ -6,8 +6,11 @@ unmasker = pipeline('fill-mask', model='dsfsi/zabantu-sot-ven-170m')
 def fill_mask(sentences):
     results = {}
     for sentence in sentences:
-        unmasked = unmasker(sentence)
-        results[sentence] = unmasked
+         if "<mask>" in sentence:
+            unmasked = unmasker(sentence)
+            results[sentence] = unmasked
+        else:
+            st.warning(f"Warning: No <mask> token found in sentence: {sentence}")
     return results
 
 def clear_text_area():
@@ -29,7 +32,7 @@ with col1:
             st.session_state.text_area = ""
     
         text_input = st.text_area(
-            "Enter sentences with <mask> token (one per line):",
+            "Enter sentences with <mask> token:",
             value=st.session_state.text_area
         )