Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
|