sonoisa commited on
Commit
ad409df
·
1 Parent(s): 35e9d5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -116,6 +116,10 @@ def preprocess_qiita_body(markdown_text):
116
  def postprocess_title(title):
117
  return re.sub(r"^title: ", "", title)
118
 
 
 
 
 
119
 
120
  trained_model, tokenizer = load_model()
121
 
@@ -124,8 +128,7 @@ USE_GPU = torch.cuda.is_available()
124
  # 推論モード設定
125
  trained_model.eval()
126
 
127
- st.title("Qiita記事タイトル案生成")
128
- st.text("記事の本文をコピペ入力して、タイトル生成ボタンを押すと、タイトル案が10個生成されます。\nGPUが使えないため生成に数十秒かかります。")
129
  body_text = st.text_area(label="記事の本文", value="", height=300, max_chars=4000)
130
  answer = st.button("タイトル生成")
131
 
 
116
  def postprocess_title(title):
117
  return re.sub(r"^title: ", "", title)
118
 
119
+ st.title("Qiita記事タイトル案生成")
120
+
121
+ description_text = st.empty()
122
+ description_text.text("...モデル読み込み中...")
123
 
124
  trained_model, tokenizer = load_model()
125
 
 
128
  # 推論モード設定
129
  trained_model.eval()
130
 
131
+ description_text.text("記事の本文をコピペ入力して、タイトル生成ボタンを押すと、タイトル案が10個生成されます。\nGPUが使えないため生成に数十秒かかります。")
 
132
  body_text = st.text_area(label="記事の本文", value="", height=300, max_chars=4000)
133
  answer = st.button("タイトル生成")
134