| import gradio as gr |
| from transformers import pipeline |
|
|
| fill_mask=pipeline("fill-mask", model="blinoff/roberta-base-russian-v0") |
|
|
| def predict_mask (sentence): |
| result=fill_mask(sentence,top_k=3) |
| outputs=[f"{r['token_str']},{round(r['score'],3)}" for r in result] |
| return '\n'.join(outputs) |
|
|
| def predict_mask1(sentence): |
| result=fill_mask(sentence,top_k=3) |
| token=[f"{r['token_str']}"for r in result] |
| score=[f"{round(r['score'],3)}" for r in result] |
| return token,score |
|
|
| demo=gr.Interface( |
| fn=predict_mask, |
| inputs=gr.Textbox(label='Текст',placeholder='Москва - <mask> России'), |
| outputs=gr.Textbox(label='Вариант заполнения'), |
| title='Ввод предложений', |
| description='Вводи предложение с маской', |
| token=[f"r['token_str']"] |
| ) |
| demo.launch(debug=True) |