sujeongim commited on
Commit
6d7b8f9
ยท
1 Parent(s): a91cc9f

edit : gr

Browse files
Files changed (1) hide show
  1. app.py +41 -2
app.py CHANGED
@@ -289,17 +289,56 @@ chatbot = gr.ChatInterface(
289
  ],
290
  )
291
 
 
 
292
  with gr.Blocks() as demo:
293
  with gr.Sidebar():
294
- # ๋กœ๊ทธ์ธ ๋ฒ„ํŠผ ์œ ์ง€ (hf_token์€ ํ˜„์žฌ ๊ตฌํ˜„์—์„œ ์‚ฌ์šฉํ•˜์ง€ ์•Š์ง€๋งŒ UI๋Š” ๊ทธ๋Œ€๋กœ ๋‘ )
295
- gr.LoginButton()
 
296
  gr.Markdown(
297
  "### ์ง„ํ–‰ ์ˆœ์„œ\n"
298
  "1) ๊ณ ์ • 10๋ฌธํ•ญ์— ๋จผ์ € ๋‹ตํ•˜๊ธฐ\n"
299
  "2) LLM์ด ์ƒ์„ฑํ•œ 20๋ฌธํ•ญ ๊ผฌ๋ฆฌ์งˆ๋ฌธ์— ๋‹ตํ•˜๊ธฐ\n"
300
  "3) ๊ณ ์ • 10๋ฌธํ•ญ์„ ๋‹ค์‹œ ํ•œ ๋ฒˆ ๋‹ตํ•˜๊ธฐ"
301
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
302
  chatbot.render()
303
 
304
  if __name__ == "__main__":
305
  demo.launch()
 
 
289
  ],
290
  )
291
 
292
+ # --- (์ƒ๋žต: ์œ„์˜ ๋ชจ๋“  import/ํ•จ์ˆ˜ ์ •์˜/์‘๋‹ต ๋กœ์ง ๋™์ผ) ---
293
+
294
  with gr.Blocks() as demo:
295
  with gr.Sidebar():
296
+ # โœ… ์ด ๋ฒ„ํŠผ์ด hf_token์„ ๊ณต๊ธ‰ํ•˜๋„๋ก ์ปดํฌ๋„ŒํŠธ๋ฅผ ๋จผ์ € ๋งŒ๋“ค์–ด ๋‘ก๋‹ˆ๋‹ค.
297
+ oauth = gr.LoginButton() # gr.OAuthButton(...)์„ ์“ฐ๋Š” ๊ฒฝ์šฐ๋„ ๋™์ผ ํฌ์ง€์…˜
298
+
299
  gr.Markdown(
300
  "### ์ง„ํ–‰ ์ˆœ์„œ\n"
301
  "1) ๊ณ ์ • 10๋ฌธํ•ญ์— ๋จผ์ € ๋‹ตํ•˜๊ธฐ\n"
302
  "2) LLM์ด ์ƒ์„ฑํ•œ 20๋ฌธํ•ญ ๊ผฌ๋ฆฌ์งˆ๋ฌธ์— ๋‹ตํ•˜๊ธฐ\n"
303
  "3) ๊ณ ์ • 10๋ฌธํ•ญ์„ ๋‹ค์‹œ ํ•œ ๋ฒˆ ๋‹ตํ•˜๊ธฐ"
304
  )
305
+
306
+ # โœ… ์ƒํƒœ ์ปดํฌ๋„ŒํŠธ๋ฅผ 'ํ•˜๋‚˜๋งŒ' ์ƒ์„ฑํ•ด ์ž…๋ ฅ/์ถœ๋ ฅ์— ์žฌ์‚ฌ์šฉ
307
+ phase = gr.State(1)
308
+ asked = gr.State(False)
309
+ i1 = gr.State(0)
310
+ i2 = gr.State(0)
311
+ i3 = gr.State(0)
312
+ gen_questions= gr.State([])
313
+ fixed_answers= gr.State([])
314
+ gen_answers = gr.State([])
315
+ rep_answers = gr.State([])
316
+
317
+ # system/max_tokens/temperature/top_p ์ž…๋ ฅ ์ปดํฌ๋„ŒํŠธ๋„ ๋ธ”๋ก ์•ˆ์—์„œ ๋งŒ๋“ค๊ณ  ๋„˜๊น๋‹ˆ๋‹ค
318
+ sys_msg = gr.Textbox(value="You are a friendly Chatbot.", label="System message")
319
+ max_toks = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
320
+ temp = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
321
+ top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
322
+
323
+ chatbot = gr.ChatInterface(
324
+ respond,
325
+ type="messages",
326
+ # โœ… ํŒŒ๋ผ๋ฏธํ„ฐ ์ˆœ์„œ: system_message, max_tokens, temperature, top_p, hf_token, (statesโ€ฆ)
327
+ additional_inputs=[
328
+ sys_msg, max_toks, temp, top_p,
329
+ oauth, # <-- โœ… hf_token ์ž๋ฆฌ์— ๋†“๊ธฐ!
330
+ phase, asked, i1, i2, i3,
331
+ gen_questions, fixed_answers, gen_answers, rep_answers,
332
+ ],
333
+ # โœ… ์ถœ๋ ฅ์—๋„ ๋™์ผํ•œ State ์ธ์Šคํ„ด์Šค ์žฌ์‚ฌ์šฉ
334
+ additional_outputs=[
335
+ phase, asked, i1, i2, i3,
336
+ gen_questions, fixed_answers, gen_answers, rep_answers,
337
+ ],
338
+ )
339
+
340
  chatbot.render()
341
 
342
  if __name__ == "__main__":
343
  demo.launch()
344
+