randmimc commited on
Commit
2a62aac
ยท
verified ยท
1 Parent(s): 2c891dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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
- summarized = pipe(prompt)
 
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()