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

edit : gr

Browse files
Files changed (1) hide show
  1. app.py +18 -56
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
- # โœ… ์ด ๋ฒ„ํŠผ์ด hf_token์„ ๊ณต๊ธ‰ํ•˜๋„๋ก ์ปดํฌ๋„ŒํŠธ๋ฅผ ๋จผ์ € ๋งŒ๋“ค์–ด ๋‘ก๋‹ˆ๋‹ค.
297
- oauth = gr.LoginButton() # gr.OAuthButton(...)์„ ์“ฐ๋Š” ๊ฒฝ์šฐ๋„ ๋™์ผ ํฌ์ง€์…˜
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
- 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โ€ฆ)
@@ -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()