Spaces:
Sleeping
Sleeping
fix: app.py
Browse files- src/app.py +2 -4
src/app.py
CHANGED
@@ -10,7 +10,7 @@ def summarize_article(input_text):
|
|
10 |
model_path = hf_hub_download(repo_id=repo_id, filename=filename)
|
11 |
|
12 |
# モデルの読み込み
|
13 |
-
llm = Llama(model_path=model_path, n_ctx=4096, n_gpu_layers=-1)
|
14 |
prompt = f"以下のテキストを日本語で約400字程度に要約してください。特に固有名詞や専門用語は正確に含めてください。テキスト: {input_text} 要約: "
|
15 |
response = llm(prompt, max_tokens=MAX_OUTPUT_TOKENS)
|
16 |
return response["choices"][0]["text"]
|
@@ -36,9 +36,7 @@ if submit_button:
|
|
36 |
summary = summarize_article(input_text)
|
37 |
|
38 |
st.markdown("## 生成された感想記事(デモ)")
|
39 |
-
st.info(
|
40 |
-
{summary["summary_text"][0]}
|
41 |
-
""")
|
42 |
|
43 |
# フッター
|
44 |
st.markdown("---")
|
|
|
10 |
model_path = hf_hub_download(repo_id=repo_id, filename=filename)
|
11 |
|
12 |
# モデルの読み込み
|
13 |
+
llm = Llama(model_path=model_path, n_ctx=4096, n_gpu_layers=-1, verbose=False)
|
14 |
prompt = f"以下のテキストを日本語で約400字程度に要約してください。特に固有名詞や専門用語は正確に含めてください。テキスト: {input_text} 要約: "
|
15 |
response = llm(prompt, max_tokens=MAX_OUTPUT_TOKENS)
|
16 |
return response["choices"][0]["text"]
|
|
|
36 |
summary = summarize_article(input_text)
|
37 |
|
38 |
st.markdown("## 生成された感想記事(デモ)")
|
39 |
+
st.info(summary)
|
|
|
|
|
40 |
|
41 |
# フッター
|
42 |
st.markdown("---")
|