ai-mask / app.py
dowiikk's picture
Create app.py
41bbe8c verified
raw
history blame contribute delete
848 Bytes
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)