Owos commited on
Commit
3854c7e
·
1 Parent(s): f82b5e5

updated requirements.txt

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -184,6 +184,7 @@ run = st.button("Get Scores")
184
  if run:
185
 
186
  progress_text = "Computing recommendation Scores"
 
187
  my_bar = st.progress(0)
188
 
189
 
@@ -199,6 +200,6 @@ if run:
199
  BATCH_SIZE = 1
200
  score = get_sense_score_batched(sentence,tokenizer,model,MLM_MASK_TOKEN,MLM_UNK_TOKEN,BATCH_SIZE)
201
  scores[model_id] = score
202
- my_bar.progress(index + 1, text=progress_text)
203
  scores = sort_dictionary(scores)
204
  st.write("Our recommendation is:", scores)
 
184
  if run:
185
 
186
  progress_text = "Computing recommendation Scores"
187
+ st.write(progress_text)
188
  my_bar = st.progress(0)
189
 
190
 
 
200
  BATCH_SIZE = 1
201
  score = get_sense_score_batched(sentence,tokenizer,model,MLM_MASK_TOKEN,MLM_UNK_TOKEN,BATCH_SIZE)
202
  scores[model_id] = score
203
+ my_bar.progress(index + 1)
204
  scores = sort_dictionary(scores)
205
  st.write("Our recommendation is:", scores)