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

Fixed error in inference.py

Browse files
Files changed (1) hide show
  1. apps/inference.py +2 -2
apps/inference.py CHANGED
@@ -61,8 +61,8 @@ def app():
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"])
66
 
67
  results_df = pd.DataFrame()
68
  results_df["Model Name"] = 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[0]["token_str"])
65
+ predicted_sentence.append(filled_sentence[0]["sequence"])
66
 
67
  results_df = pd.DataFrame()
68
  results_df["Model Name"] = models