chEstyleU / app.py
masa729406's picture
Update app.py
bd0cb89
raw
history blame
367 Bytes
!pip install transformers
import transformers
# GPT-2ヒデルをθͺ­γΏθΎΌγ‚€
model = transformers.pipeline("text-generation", model="gpt2")
# γƒ†γ‚­γ‚Ήγƒˆγ‚’η”Ÿζˆγ™γ‚‹
generated_text = model("ァンガスタジをム", max_length=50)[0]['generated_text']
import streamlit as st
# UIγ‚’δ½œζˆγ™γ‚‹
st.title("ァンガスタジをム")
st.write(generated_text)