| import streamlit as st |
| from transformers import pipeline |
|
|
| option = st.selectbox( |
| 'Choisissez votre langue / 言語を選択', |
| ('Français', '日本語')) |
|
|
| if option == "Français": |
| pipeline = pipeline(task="text-generation", model="Aruno/Bloom-FR-160m") |
| title_str = "Génération de texte en français" |
| min_length_str = "Longueur minimum" |
| max_length_str = "Longueur maximum" |
| num_beams_str = "Nombre de beam" |
| num_output_str = "Nombre de réponse" |
| submitted_str = "Génération" |
| input_value_str1 = 'Entrée' |
| input_value_str2 = 'Je vais dans' |
| else: |
| pipeline = pipeline(task="text-generation", model="Aruno/Bloom-JP-160m") |
| title_str = "日本語のテキスト生成" |
| min_length_str = "生成最小数" |
| max_length_str = "生成最大数" |
| num_beams_str = "Beam数" |
| num_output_str = "応答数" |
| submitted_str = "生成" |
| input_value_str1 = '入力' |
| input_value_str2 = '宇宙に行って、' |
|
|
| st.title(title_str) |
|
|
| with st.form("my_form"): |
| input_value = st.text_input(input_value_str1, input_value_str2) |
| min_length = st.slider(min_length_str, min_value=1, value=16) |
| max_length = st.slider(max_length_str, min_value=1, value=32) |
| num_beams = st.slider(num_beams_str, min_value=1, max_value=10, value=3) |
| num_output = st.slider(num_output_str, min_value=1, max_value=10, value=3) |
|
|
| submitted = st.form_submit_button(submitted_str) |
| if submitted: |
| predictions = pipeline(input_value, min_length=min_length, max_length=max_length, num_beams=num_beams, num_return_sequences=num_output) |
| st.write(predictions) |