prateekagrawal commited on
Commit
92e58bb
1 Parent(s): 8754e0c

Updated inference.py

Browse files
Files changed (1) hide show
  1. apps/inference.py +1 -0
apps/inference.py CHANGED
@@ -59,6 +59,7 @@ def app():
59
 
60
  for i in range(len(selected_models)):
61
  filled_sentence = load_model(masked_text, selected_models[i])
 
62
  models.append(selected_models[i])
63
  predicted_tokens.append(filled_sentence["token_str"])
64
  predicted_sentence.append(filled_sentence["sequence"])
 
59
 
60
  for i in range(len(selected_models)):
61
  filled_sentence = load_model(masked_text, selected_models[i])
62
+ st.write(filled_sentence)
63
  models.append(selected_models[i])
64
  predicted_tokens.append(filled_sentence["token_str"])
65
  predicted_sentence.append(filled_sentence["sequence"])