modify comment
Browse files
app.py
CHANGED
|
@@ -58,9 +58,6 @@ def summy(text):
|
|
| 58 |
Returns
|
| 59 |
summarize_text: str
|
| 60 |
要約結果のテキスト
|
| 61 |
-
|
| 62 |
-
TODO
|
| 63 |
-
処理の実装
|
| 64 |
"""
|
| 65 |
inputs = tokenizer_sum("summarize: " + text, return_tensors="pt")
|
| 66 |
outputs = model_sum.generate(
|
|
@@ -85,9 +82,6 @@ def generate_questions(answer_1, answer_2, text):
|
|
| 85 |
Returns
|
| 86 |
generated_questions: list[str]
|
| 87 |
生成された質問文のリスト
|
| 88 |
-
|
| 89 |
-
TODO
|
| 90 |
-
処理の実装
|
| 91 |
"""
|
| 92 |
answer_context_list = [(answer, text) for answer in [answer_1, answer_2]] # 解答を質問生成する元となる文(要約結果)とセットにする。
|
| 93 |
generated_questions = []
|
|
@@ -123,8 +117,6 @@ def extract_answer(question, text):
|
|
| 123 |
answer: str
|
| 124 |
回答のテキスト
|
| 125 |
|
| 126 |
-
TODO
|
| 127 |
-
処理の実装
|
| 128 |
"""
|
| 129 |
inputs = tokenizer_qa(question, text, return_tensors="pt") # tokenizerには複数のテキストを与える
|
| 130 |
|
|
@@ -146,8 +138,6 @@ def extract_answer(question, text):
|
|
| 146 |
|
| 147 |
def extract_answer_all(gen_q_1, gen_q_2, source_text, sum_text):
|
| 148 |
"""extract_answer()をまとめて実行する
|
| 149 |
-
TODO
|
| 150 |
-
処理の実装
|
| 151 |
"""
|
| 152 |
a_source_1 = extract_answer(gen_q_1, source_text)
|
| 153 |
a_sum_1 = extract_answer(gen_q_1, sum_text)
|
|
@@ -159,13 +149,13 @@ def extract_answer_all(gen_q_1, gen_q_2, source_text, sum_text):
|
|
| 159 |
# 2. UIの定義
|
| 160 |
with gr.Blocks() as demo:
|
| 161 |
gr.Markdown("### 1. 要約生成")
|
| 162 |
-
#
|
| 163 |
source_text = gr.Textbox(label="要約対象")
|
| 164 |
btn_summy = gr.Button("要約生成")
|
| 165 |
sum_text = gr.Textbox(label="要約結果")
|
| 166 |
|
| 167 |
gr.Markdown("### 2. 質問生成")
|
| 168 |
-
#
|
| 169 |
|
| 170 |
with gr.Row():
|
| 171 |
with gr.Column():
|
|
|
|
| 58 |
Returns
|
| 59 |
summarize_text: str
|
| 60 |
要約結果のテキスト
|
|
|
|
|
|
|
|
|
|
| 61 |
"""
|
| 62 |
inputs = tokenizer_sum("summarize: " + text, return_tensors="pt")
|
| 63 |
outputs = model_sum.generate(
|
|
|
|
| 82 |
Returns
|
| 83 |
generated_questions: list[str]
|
| 84 |
生成された質問文のリスト
|
|
|
|
|
|
|
|
|
|
| 85 |
"""
|
| 86 |
answer_context_list = [(answer, text) for answer in [answer_1, answer_2]] # 解答を質問生成する元となる文(要約結果)とセットにする。
|
| 87 |
generated_questions = []
|
|
|
|
| 117 |
answer: str
|
| 118 |
回答のテキスト
|
| 119 |
|
|
|
|
|
|
|
| 120 |
"""
|
| 121 |
inputs = tokenizer_qa(question, text, return_tensors="pt") # tokenizerには複数のテキストを与える
|
| 122 |
|
|
|
|
| 138 |
|
| 139 |
def extract_answer_all(gen_q_1, gen_q_2, source_text, sum_text):
|
| 140 |
"""extract_answer()をまとめて実行する
|
|
|
|
|
|
|
| 141 |
"""
|
| 142 |
a_source_1 = extract_answer(gen_q_1, source_text)
|
| 143 |
a_sum_1 = extract_answer(gen_q_1, sum_text)
|
|
|
|
| 149 |
# 2. UIの定義
|
| 150 |
with gr.Blocks() as demo:
|
| 151 |
gr.Markdown("### 1. 要約生成")
|
| 152 |
+
# 要約のための入出力UIの作成
|
| 153 |
source_text = gr.Textbox(label="要約対象")
|
| 154 |
btn_summy = gr.Button("要約生成")
|
| 155 |
sum_text = gr.Textbox(label="要約結果")
|
| 156 |
|
| 157 |
gr.Markdown("### 2. 質問生成")
|
| 158 |
+
# 質問文生成のための入力UIの作成
|
| 159 |
|
| 160 |
with gr.Row():
|
| 161 |
with gr.Column():
|