Aruno commited on
Commit
a3b9f76
1 Parent(s): 0996a0b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ pipeline = pipeline(task="text-generation", model="Aruno/Bloom-JP-160m")
5
+
6
+ st.title("日本語のテキスト生成")
7
+
8
+ with st.form("my_form"):
9
+ title = st.text_input('入力', '宇宙に行って、')
10
+ min_length = st.slider("生成最小数", min_value=1, value=16)
11
+ max_length = st.slider("生成最大数", min_value=1, value=32)
12
+ num_beams = st.slider("Beam数",min_value=1, max_value=10, value=3)
13
+ num_output = st.slider("応答数",min_value=1, max_value=10, value=3)
14
+
15
+ submitted = st.form_submit_button("生成")
16
+ if submitted:
17
+ predictions = pipeline(title, min_length=min_length, max_length=max_length, num_beams=num_beams, num_return_sequences=num_output)
18
+ st.write(predictions)