Spaces:
Sleeping
Sleeping
sujeongim commited on
Commit ยท
01dfce8
1
Parent(s): 6d7b8f9
edit : gr
Browse files
app.py
CHANGED
|
@@ -255,47 +255,12 @@ def respond(
|
|
| 255 |
# --------------------------------
|
| 256 |
# 4) ChatInterface ๊ตฌ์ฑ
|
| 257 |
# --------------------------------
|
| 258 |
-
chatbot = gr.ChatInterface(
|
| 259 |
-
respond,
|
| 260 |
-
type="messages",
|
| 261 |
-
additional_inputs=[
|
| 262 |
-
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
| 263 |
-
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
| 264 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
| 265 |
-
gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
|
| 266 |
-
# --- ์ํ๋ค ---
|
| 267 |
-
gr.State(value=1), # phase
|
| 268 |
-
gr.State(value=False), # asked
|
| 269 |
-
gr.State(value=0), # i1
|
| 270 |
-
gr.State(value=0), # i2
|
| 271 |
-
gr.State(value=0), # i3
|
| 272 |
-
gr.State(value=[]), # gen_questions (list[str])
|
| 273 |
-
gr.State(value=[]), # fixed_answers (list[str])
|
| 274 |
-
gr.State(value=[]), # gen_answers (list[str])
|
| 275 |
-
gr.State(value=[]), # rep_answers (list[str])
|
| 276 |
-
],
|
| 277 |
-
# respond๊ฐ ์คํธ๋ฆฌ๋ฐ์ผ๋ก (assistant_text, ์ํ๋ค...)์ yield ํ๋ฏ๋ก,
|
| 278 |
-
# ์๋ additional_outputs ์์/๊ฐ์๋ inputs์ ์ ํํ ์ผ์นํด์ผ ํจ
|
| 279 |
-
additional_outputs=[
|
| 280 |
-
gr.State(), # phase
|
| 281 |
-
gr.State(), # asked
|
| 282 |
-
gr.State(), # i1
|
| 283 |
-
gr.State(), # i2
|
| 284 |
-
gr.State(), # i3
|
| 285 |
-
gr.State(), # gen_questions
|
| 286 |
-
gr.State(), # fixed_answers
|
| 287 |
-
gr.State(), # gen_answers
|
| 288 |
-
gr.State(), # rep_answers
|
| 289 |
-
],
|
| 290 |
-
)
|
| 291 |
|
| 292 |
-
# --- (์๋ต: ์์ ๋ชจ๋ import/ํจ์ ์ ์/์๋ต ๋ก์ง ๋์ผ) ---
|
| 293 |
|
| 294 |
with gr.Blocks() as demo:
|
| 295 |
with gr.Sidebar():
|
| 296 |
-
#
|
| 297 |
-
oauth = gr.LoginButton()
|
| 298 |
-
|
| 299 |
gr.Markdown(
|
| 300 |
"### ์งํ ์์\n"
|
| 301 |
"1) ๊ณ ์ 10๋ฌธํญ์ ๋จผ์ ๋ตํ๊ธฐ\n"
|
|
@@ -303,24 +268,23 @@ with gr.Blocks() as demo:
|
|
| 303 |
"3) ๊ณ ์ 10๋ฌธํญ์ ๋ค์ ํ ๋ฒ ๋ตํ๊ธฐ"
|
| 304 |
)
|
| 305 |
|
| 306 |
-
|
| 307 |
-
|
| 308 |
-
|
| 309 |
-
|
| 310 |
-
|
| 311 |
-
|
| 312 |
-
|
| 313 |
-
|
| 314 |
-
|
| 315 |
-
rep_answers = gr.State([])
|
| 316 |
|
| 317 |
-
|
| 318 |
-
|
| 319 |
-
|
| 320 |
-
|
| 321 |
-
|
| 322 |
|
| 323 |
-
|
| 324 |
respond,
|
| 325 |
type="messages",
|
| 326 |
# โ
ํ๋ผ๋ฏธํฐ ์์: system_message, max_tokens, temperature, top_p, hf_token, (statesโฆ)
|
|
@@ -336,9 +300,7 @@ with gr.Blocks() as demo:
|
|
| 336 |
gen_questions, fixed_answers, gen_answers, rep_answers,
|
| 337 |
],
|
| 338 |
)
|
| 339 |
-
|
| 340 |
-
chatbot.render()
|
| 341 |
|
| 342 |
if __name__ == "__main__":
|
| 343 |
demo.launch()
|
| 344 |
-
|
|
|
|
| 255 |
# --------------------------------
|
| 256 |
# 4) ChatInterface ๊ตฌ์ฑ
|
| 257 |
# --------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 258 |
|
|
|
|
| 259 |
|
| 260 |
with gr.Blocks() as demo:
|
| 261 |
with gr.Sidebar():
|
| 262 |
+
# ๋ก๊ทธ์ธ ๋ฒํผ ์ ์ง (hf_token์ ํ์ฌ ๊ตฌํ์์ ์ฌ์ฉํ์ง ์์ง๋ง UI๋ ๊ทธ๋๋ก ๋ )
|
| 263 |
+
oauth = gr.LoginButton()
|
|
|
|
| 264 |
gr.Markdown(
|
| 265 |
"### ์งํ ์์\n"
|
| 266 |
"1) ๊ณ ์ 10๋ฌธํญ์ ๋จผ์ ๋ตํ๊ธฐ\n"
|
|
|
|
| 268 |
"3) ๊ณ ์ 10๋ฌธํญ์ ๋ค์ ํ ๋ฒ ๋ตํ๊ธฐ"
|
| 269 |
)
|
| 270 |
|
| 271 |
+
phase = gr.State(1)
|
| 272 |
+
asked = gr.State(False)
|
| 273 |
+
i1 = gr.State(0)
|
| 274 |
+
i2 = gr.State(0)
|
| 275 |
+
i3 = gr.State(0)
|
| 276 |
+
gen_questions= gr.State([])
|
| 277 |
+
fixed_answers= gr.State([])
|
| 278 |
+
gen_answers = gr.State([])
|
| 279 |
+
rep_answers = gr.State([])
|
|
|
|
| 280 |
|
| 281 |
+
# system/max_tokens/temperature/top_p ์
๋ ฅ ์ปดํฌ๋ํธ๋ ๋ธ๋ก ์์์ ๋ง๋ค๊ณ ๋๊น๋๋ค
|
| 282 |
+
sys_msg = gr.Textbox(value="You are a friendly Chatbot.", label="System message")
|
| 283 |
+
max_toks = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
|
| 284 |
+
temp = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
|
| 285 |
+
top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
|
| 286 |
|
| 287 |
+
chatbot = gr.ChatInterface(
|
| 288 |
respond,
|
| 289 |
type="messages",
|
| 290 |
# โ
ํ๋ผ๋ฏธํฐ ์์: system_message, max_tokens, temperature, top_p, hf_token, (statesโฆ)
|
|
|
|
| 300 |
gen_questions, fixed_answers, gen_answers, rep_answers,
|
| 301 |
],
|
| 302 |
)
|
| 303 |
+
chatbot.render()
|
|
|
|
| 304 |
|
| 305 |
if __name__ == "__main__":
|
| 306 |
demo.launch()
|
|
|