Update app.py

app.py

@@ -13,6 +13,11 @@ def fill_mask(sentences):
     for key, (language, sentence) in sentences.items():
         if language == 'choose language':
             warnings.append(f"Warning: Choose language for {sentence}")
+            continue
+            
+        if language != 'choose language' and sentence == ''
+            warnings.append(f"Warning: Enter sentence for {language}")
+            continue
         
         if "<mask>" in sentence:
             masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
@@ -38,6 +43,7 @@ if 'text_input' not in st.session_state:
 if 'warnings' not in st.session_state:
     st.session_state['warnings'] = []
 
+input_sentences = {}
 with col1:
     with st.container(border=True):
         st.markdown("Input :clipboard:")
@@ -49,7 +55,6 @@ with col1:
         language_options = ['Choose language', 'Zulu', 'Tshivenda', 'Sepedi', 'Tswana', 'Tsonga']
         
         option_selected = st.selectbox(f"Select an input option:", select_options, index=0)
-        input_sentences = {}
         
         if option_selected == 'Enter text input':
             st.session_state['warnings'].clear()
@@ -153,7 +158,6 @@ with col2:
 if 'result' in locals():  
     if result:
         line = 0
-        st.write(f"{result.items()}")
         for sentence, predictions in result.items():
             line += 1
             predicted_word = predictions[0]['token_str']