UnarineLeo commited on
Commit
4199132
1 Parent(s): b90e99a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def fill_mask(sentences):
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:
@@ -153,6 +153,7 @@ with col2:
153
  if 'result' in locals():
154
  if result:
155
  line = 0
 
156
  for sentence, predictions in result.items():
157
  line += 1
158
  predicted_word = predictions[0]['token_str']
 
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:
 
153
  if 'result' in locals():
154
  if result:
155
  line = 0
156
+ st.write(f"{result.items()}")
157
  for sentence, predictions in result.items():
158
  line += 1
159
  predicted_word = predictions[0]['token_str']