Spaces:
Runtime error
Runtime error
Commit
•
e668d73
1
Parent(s):
92e58bb
Fixed error in inference.py
Browse files- 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
|