Update app.py
Browse files
app.py
CHANGED
|
@@ -22,9 +22,9 @@ def respond(
|
|
| 22 |
message,
|
| 23 |
history: list[tuple[str, str]],
|
| 24 |
system_message="", # ๊ธฐ๋ณธ๊ฐ ์ถ๊ฐ
|
| 25 |
-
max_tokens=1024,
|
| 26 |
-
temperature=0.7,
|
| 27 |
-
top_p=0.9,
|
| 28 |
):
|
| 29 |
global fashion_code
|
| 30 |
system_prefix = """๋ฐ๋์ ํ๊ธ๋ก ๋ต๋ณํ ๊ฒ. ๋๋ ์ฃผ์ด์ง ์์ค์ฝ๋๋ฅผ ๊ธฐ๋ฐ์ผ๋ก "์๋น์ค ์ฌ์ฉ ์ค๋ช
๋ฐ ์๋ด, qna๋ฅผ ํ๋ ์ญํ ์ด๋ค". ์์ฃผ ์น์ ํ๊ณ ์์ธํ๊ฒ 4000ํ ํฐ ์ด์ ์์ฑํ๋ผ. ๋๋ ์ฝ๋๋ฅผ ๊ธฐ๋ฐ์ผ๋ก ์ฌ์ฉ ์ค๋ช
๋ฐ ์ง์ ์๋ต์ ์งํํ๋ฉฐ, ์ด์ฉ์์๊ฒ ๋์์ ์ฃผ์ด์ผ ํ๋ค. ์ด์ฉ์๊ฐ ๊ถ๊ธํด ํ ๋ง ํ ๋ด์ฉ์ ์น์ ํ๊ฒ ์๋ ค์ฃผ๋๋ก ํ๋ผ. ์ฝ๋ ์ ์ฒด ๋ด์ฉ์ ๋ํด์๋ ๋ณด์์ ์ ์งํ๊ณ , ํค ๊ฐ ๋ฐ ์๋ํฌ์ธํธ์ ๊ตฌ์ฒด์ ์ธ ๋ชจ๋ธ์ ๊ณต๊ฐํ์ง ๋ง๋ผ. """
|
|
@@ -34,15 +34,12 @@ def respond(
|
|
| 34 |
system_message += f"\n\nํจ์
์ฝ๋ ๋ด์ฉ:\n{fashion_code}"
|
| 35 |
message = "ํจ์
์ฝ๋์ ๋ํด ์ค๋ช
ํด์ฃผ์ธ์."
|
| 36 |
|
| 37 |
-
messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}]
|
| 38 |
-
|
| 39 |
-
# prefix ์ถ๊ฐ
|
| 40 |
for val in history:
|
| 41 |
if val[0]:
|
| 42 |
messages.append({"role": "user", "content": val[0]})
|
| 43 |
if val[1]:
|
| 44 |
messages.append({"role": "assistant", "content": val[1]})
|
| 45 |
-
|
| 46 |
messages.append({"role": "user", "content": message})
|
| 47 |
|
| 48 |
response = ""
|
|
@@ -56,7 +53,7 @@ def respond(
|
|
| 56 |
token = message.choices[0].delta.content
|
| 57 |
if token is not None:
|
| 58 |
response += token.strip("") # ํ ํฐ ์ ๊ฑฐ
|
| 59 |
-
|
| 60 |
|
| 61 |
# Gradio ์ธํฐํ์ด์ค ์ค์ ๋ถ๋ถ๋ ์์
|
| 62 |
demo = gr.ChatInterface(
|
|
@@ -68,14 +65,14 @@ demo = gr.ChatInterface(
|
|
| 68 |
gr.Slider(minimum=0, maximum=1, value=0.9, label="Top P"),
|
| 69 |
],
|
| 70 |
examples=[
|
| 71 |
-
["ํจ์
์ฝ๋ ์คํ"],
|
| 72 |
-
["์ฌ์ฉ ๋ฐฉ๋ฒ์ ์์ธํ ์ค๋ช
ํ๋ผ"],
|
| 73 |
-
["์ฌ์ฉ ๋ฐฉ๋ฒ์ ์ ํ๋ธ ์์ ์คํฌ๋ฆฝํธ ํํ๋ก ์์ฑํ๋ผ"],
|
| 74 |
["์ฌ์ฉ ๋ฐฉ๋ฒ์ SEO ์ต์ ํํ์ฌ ๋ธ๋ก๊ทธ ํฌ์คํธ๋ก 4000 ํ ํฐ ์ด์ ์์ฑํ๋ผ"],
|
| 75 |
["๊ณ์ ์ด์ด์ ๋ต๋ณํ๋ผ"],
|
| 76 |
],
|
| 77 |
cache_examples=False, # ์บ์ฑ ๋นํ์ฑํ ์ค์
|
| 78 |
-
# css="""footer {visibility: hidden}""",
|
| 79 |
)
|
| 80 |
|
| 81 |
if __name__ == "__main__":
|
|
|
|
| 22 |
message,
|
| 23 |
history: list[tuple[str, str]],
|
| 24 |
system_message="", # ๊ธฐ๋ณธ๊ฐ ์ถ๊ฐ
|
| 25 |
+
max_tokens=1024, # ๊ธฐ๋ณธ๊ฐ ์ถ๊ฐ
|
| 26 |
+
temperature=0.7, # ๊ธฐ๋ณธ๊ฐ ์ถ๊ฐ
|
| 27 |
+
top_p=0.9, # ๊ธฐ๋ณธ๊ฐ ์ถ๊ฐ
|
| 28 |
):
|
| 29 |
global fashion_code
|
| 30 |
system_prefix = """๋ฐ๋์ ํ๊ธ๋ก ๋ต๋ณํ ๊ฒ. ๋๋ ์ฃผ์ด์ง ์์ค์ฝ๋๋ฅผ ๊ธฐ๋ฐ์ผ๋ก "์๋น์ค ์ฌ์ฉ ์ค๋ช
๋ฐ ์๋ด, qna๋ฅผ ํ๋ ์ญํ ์ด๋ค". ์์ฃผ ์น์ ํ๊ณ ์์ธํ๊ฒ 4000ํ ํฐ ์ด์ ์์ฑํ๋ผ. ๋๋ ์ฝ๋๋ฅผ ๊ธฐ๋ฐ์ผ๋ก ์ฌ์ฉ ์ค๋ช
๋ฐ ์ง์ ์๋ต์ ์งํํ๋ฉฐ, ์ด์ฉ์์๊ฒ ๋์์ ์ฃผ์ด์ผ ํ๋ค. ์ด์ฉ์๊ฐ ๊ถ๊ธํด ํ ๋ง ํ ๋ด์ฉ์ ์น์ ํ๊ฒ ์๋ ค์ฃผ๋๋ก ํ๋ผ. ์ฝ๋ ์ ์ฒด ๋ด์ฉ์ ๋ํด์๋ ๋ณด์์ ์ ์งํ๊ณ , ํค ๊ฐ ๋ฐ ์๋ํฌ์ธํธ์ ๊ตฌ์ฒด์ ์ธ ๋ชจ๋ธ์ ๊ณต๊ฐํ์ง ๋ง๋ผ. """
|
|
|
|
| 34 |
system_message += f"\n\nํจ์
์ฝ๋ ๋ด์ฉ:\n{fashion_code}"
|
| 35 |
message = "ํจ์
์ฝ๋์ ๋ํด ์ค๋ช
ํด์ฃผ์ธ์."
|
| 36 |
|
| 37 |
+
messages = [{"role": "system", "content": f"{system_prefix} {system_message}"}] # prefix ์ถ๊ฐ
|
|
|
|
|
|
|
| 38 |
for val in history:
|
| 39 |
if val[0]:
|
| 40 |
messages.append({"role": "user", "content": val[0]})
|
| 41 |
if val[1]:
|
| 42 |
messages.append({"role": "assistant", "content": val[1]})
|
|
|
|
| 43 |
messages.append({"role": "user", "content": message})
|
| 44 |
|
| 45 |
response = ""
|
|
|
|
| 53 |
token = message.choices[0].delta.content
|
| 54 |
if token is not None:
|
| 55 |
response += token.strip("") # ํ ํฐ ์ ๊ฑฐ
|
| 56 |
+
yield response
|
| 57 |
|
| 58 |
# Gradio ์ธํฐํ์ด์ค ์ค์ ๋ถ๋ถ๋ ์์
|
| 59 |
demo = gr.ChatInterface(
|
|
|
|
| 65 |
gr.Slider(minimum=0, maximum=1, value=0.9, label="Top P"),
|
| 66 |
],
|
| 67 |
examples=[
|
| 68 |
+
["ํจ์
์ฝ๋ ์คํ"],
|
| 69 |
+
["์ฌ์ฉ ๋ฐฉ๋ฒ์ 4000 ํ ํฐ ์ด์ ์์ธํ ์ค๋ช
ํ๋ผ"],
|
| 70 |
+
["์ฌ์ฉ ๋ฐฉ๋ฒ์ 4000 ํ ํฐ ์ด์ ์ ํ๋ธ ์์ ์คํฌ๋ฆฝํธ ํํ๋ก ์์ฑํ๋ผ"],
|
| 71 |
["์ฌ์ฉ ๋ฐฉ๋ฒ์ SEO ์ต์ ํํ์ฌ ๋ธ๋ก๊ทธ ํฌ์คํธ๋ก 4000 ํ ํฐ ์ด์ ์์ฑํ๋ผ"],
|
| 72 |
["๊ณ์ ์ด์ด์ ๋ต๋ณํ๋ผ"],
|
| 73 |
],
|
| 74 |
cache_examples=False, # ์บ์ฑ ๋นํ์ฑํ ์ค์
|
| 75 |
+
# css="""footer {visibility: hidden}""", # ์ด๊ณณ์ CSS๋ฅผ ์ถ๊ฐ
|
| 76 |
)
|
| 77 |
|
| 78 |
if __name__ == "__main__":
|