Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,6 @@ model.eval()
|
|
| 9 |
pipeline = FillMaskPipeline(model, tokenizer_hf)
|
| 10 |
text = f"Em dic <mask>."
|
| 11 |
res_hf = pipeline(text)
|
| 12 |
-
|
| 13 |
|
| 14 |
-
st.write(
|
|
|
|
| 9 |
pipeline = FillMaskPipeline(model, tokenizer_hf)
|
| 10 |
text = f"Em dic <mask>."
|
| 11 |
res_hf = pipeline(text)
|
| 12 |
+
res = ([r['token_str'] for r in res_hf])
|
| 13 |
|
| 14 |
+
st.write(res)
|