dowiikk commited on
Commit
41bbe8c
·
verified ·
1 Parent(s): 29002cc

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ fill_mask=pipeline("fill-mask", model="blinoff/roberta-base-russian-v0")
5
+
6
+ def predict_mask (sentence):
7
+ result=fill_mask(sentence,top_k=3)
8
+ outputs=[f"{r['token_str']},{round(r['score'],3)}" for r in result]
9
+ return '\n'.join(outputs)
10
+
11
+ def predict_mask1(sentence):
12
+ result=fill_mask(sentence,top_k=3)
13
+ token=[f"{r['token_str']}"for r in result]
14
+ score=[f"{round(r['score'],3)}" for r in result]
15
+ return token,score
16
+
17
+ demo=gr.Interface(
18
+ fn=predict_mask,
19
+ inputs=gr.Textbox(label='Текст',placeholder='Москва - <mask> России'),
20
+ outputs=gr.Textbox(label='Вариант заполнения'),
21
+ title='Ввод предложений',
22
+ description='Вводи предложение с маской',
23
+ token=[f"r['token_str']"]
24
+ )
25
+ demo.launch(debug=True)