| import streamlit as st | |
| from transformers import AutoModelForMaskedLM | |
| from transformers import AutoTokenizer, FillMaskPipeline | |
| from pprint import pprint | |
| tokenizer_hf = AutoTokenizer.from_pretrained('projecte-aina/roberta-base-ca-v2') | |
| model = AutoModelForMaskedLM.from_pretrained('projecte-aina/roberta-base-ca-v2') | |
| model.eval() | |
| pipeline = FillMaskPipeline(model, tokenizer_hf) | |
| text = f"Tinc pressa, marxo al MACBA a veure l'exposició <mask>." | |
| res_hf = pipeline(text) | |
| res = ([r['token_str'] for r in res_hf]) | |
| st.write(res) |