Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,5 +1,9 @@
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
|
| 4 |
pipe = pipeline("summarization", model="gangyeolkim/kobart-korean-summarizer-v2")
|
| 5 |
|
|
@@ -78,7 +82,7 @@ if not st.session_state.messages:
|
|
| 78 |
ํ์ง๋ง ! ๊ฐ๋จํ ํ์ฉํด ๋ณด์ค ์ ์๋ ์์ฝ ๋ชจ๋ธ์ ์ ๊ณตํ๊ณ ์์ต๋๋ค.
|
| 79 |
์งง์ ์ ๋ฌธ๊ธฐ์ฌ๋ฅผ ๋งํด ์ฃผ์๋ฉด ์์ฝ๋ฌธ์ ์ ๊ณตํฉ๋ค.
|
| 80 |
๊ณ์ ์ฌ์ฉํ๊ณ ์ถ๋ค๋ฉด "yes" ๋ฅผ ์
๋ ฅํด์ฃผ์ธ์.
|
| 81 |
-
"""
|
| 82 |
st.rerun()
|
| 83 |
|
| 84 |
if prompt := st.chat_input():
|
|
@@ -91,7 +95,8 @@ if prompt := st.chat_input():
|
|
| 91 |
st.session_state.messages.append({"role": "assistant", "avatar": assistant_icon์'})
|
| 92 |
|
| 93 |
elif st.session_state.approval_state == "approved":
|
| 94 |
-
|
|
|
|
| 95 |
st.session_state.messages.append({"role": "assistant", "avatar": assistant_icon_path, "content": summarized[0]["summary_text"]})
|
| 96 |
|
| 97 |
-
st.rerun()
|
|
|
|
| 1 |
import streamlit as st
|
| 2 |
from transformers import pipeline
|
| 3 |
+
import re
|
| 4 |
+
|
| 5 |
+
def replace_multiple_spaces(text):
|
| 6 |
+
return re.sub(r'\s+', ' ', text)
|
| 7 |
|
| 8 |
pipe = pipeline("summarization", model="gangyeolkim/kobart-korean-summarizer-v2")
|
| 9 |
|
|
|
|
| 82 |
ํ์ง๋ง ! ๊ฐ๋จํ ํ์ฉํด ๋ณด์ค ์ ์๋ ์์ฝ ๋ชจ๋ธ์ ์ ๊ณตํ๊ณ ์์ต๋๋ค.
|
| 83 |
์งง์ ์ ๋ฌธ๊ธฐ์ฌ๋ฅผ ๋งํด ์ฃผ์๋ฉด ์์ฝ๋ฌธ์ ์ ๊ณตํฉ๋ค.
|
| 84 |
๊ณ์ ์ฌ์ฉํ๊ณ ์ถ๋ค๋ฉด "yes" ๋ฅผ ์
๋ ฅํด์ฃผ์ธ์.
|
| 85 |
+
"""})
|
| 86 |
st.rerun()
|
| 87 |
|
| 88 |
if prompt := st.chat_input():
|
|
|
|
| 95 |
st.session_state.messages.append({"role": "assistant", "avatar": assistant_icon์'})
|
| 96 |
|
| 97 |
elif st.session_state.approval_state == "approved":
|
| 98 |
+
result = replace_multiple_spaces(prompt)
|
| 99 |
+
summarized = pipe(result)
|
| 100 |
st.session_state.messages.append({"role": "assistant", "avatar": assistant_icon_path, "content": summarized[0]["summary_text"]})
|
| 101 |
|
| 102 |
+
# st.rerun()
|