UnarineLeo commited on
Commit
b90e99a
1 Parent(s): 8d1b776

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -9,9 +9,11 @@ st.set_page_config(layout="wide")
9
  def fill_mask(sentences):
10
  results = {}
11
  warnings = []
12
- warnings.append(f"= {sentences.items()}")
13
  for key, (language, sentence) in sentences.items():
14
- warnings.append(f"=> {sentence}")
 
 
15
  if "<mask>" in sentence:
16
  masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
17
  unmasked = unmasker(masked_sentence)
 
9
  def fill_mask(sentences):
10
  results = {}
11
  warnings = []
12
+ # warnings.append(f"= {sentences.items()}")
13
  for key, (language, sentence) in sentences.items():
14
+ if language == 'choose language'
15
+ warnings.append(f"Warning: Choose language for {sentence}")
16
+
17
  if "<mask>" in sentence:
18
  masked_sentence = sentence.replace('<mask>', unmasker.tokenizer.mask_token)
19
  unmasked = unmasker(masked_sentence)