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
|