taishi-i commited on
Commit
e1188cc
β€’
1 Parent(s): 57dc917

add app.py and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +30 -0
  2. requirements.txt +3 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+
3
+ from transformers import pipeline
4
+ from nagisa_bert import NagisaBertTokenizer
5
+
6
+ tokenizer = NagisaBertTokenizer.from_pretrained("taishi-i/nagisa_bert")
7
+ fill_mask = pipeline(
8
+ "fill-mask",
9
+ model="taishi-i/nagisa_bert",
10
+ tokenizer=tokenizer
11
+ )
12
+
13
+ with st.form("Fill-mask"):
14
+ text = st.text_area(
15
+ label="[MASK]γ‚’ε«γ‚€γƒ†γ‚­γ‚Ήγƒˆγ‚’ε…₯εŠ›γ—γ¦γγ γ•γ„γ€‚",
16
+ value="nagisaで[MASK]できるヒデルです",
17
+ max_chars=512
18
+ )
19
+ submitted = st.form_submit_button("Submit")
20
+
21
+ if text:
22
+ out = fill_mask(text)
23
+ st.json(out)
24
+
25
+ if submitted:
26
+ if "[MASK]" in text:
27
+ out = fill_mask(text)
28
+ st.json(out)
29
+ else:
30
+ st.write("[MASK] γ‚’ε«γ‚€γƒ†γ‚­γ‚Ήγƒˆγ‚’ε…₯εŠ›γ—γ¦γγ γ•γ„γ€‚")
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ torch
2
+ transformers
3
+ nagisa_bert