Kims12 commited on
Commit
e3e9a1c
·
verified ·
1 Parent(s): a52cf12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +70 -60
app.py CHANGED
@@ -156,6 +156,7 @@ with gr.Blocks() as demo:
156
  gr.Markdown("# Prompting AI - 일반 문답형 데모")
157
  gr.Markdown("언어모델별 문답형 테스트 데모입니다. 한 번에 한 질문씩만 주고받습니다.")
158
 
 
159
  with gr.Row():
160
  hf_token_box = gr.Textbox(
161
  label="HuggingFace 토큰 (비공개)",
@@ -167,6 +168,14 @@ with gr.Blocks() as demo:
167
  type="password",
168
  placeholder="OpenAI API 토큰을 입력하세요..."
169
  )
 
 
 
 
 
 
 
 
170
 
171
  # --- Tab: 일반 모델 ---
172
  with gr.Tab("일반 모델"):
@@ -175,17 +184,20 @@ with gr.Blocks() as demo:
175
  label="Language Model (HuggingFace)",
176
  value="Zephyr 7B Beta"
177
  )
178
- max_tokens = gr.Slider(minimum=0, maximum=2000, value=500, step=100, label="Max Tokens")
179
- temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
180
- top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
181
- system_message = gr.Textbox(
182
- value="""반드시 한글로 답변할 것.
 
 
 
183
  너는 최고의 비서이다.
184
  내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.
185
  """,
186
- label="System Message",
187
- lines=3
188
- )
189
 
190
  question_input = gr.Textbox(label="질문을 입력하세요")
191
  answer_output = gr.Textbox(label="답변", interactive=False)
@@ -208,17 +220,19 @@ with gr.Blocks() as demo:
208
 
209
  # --- Tab: Cohere Command R+ ---
210
  with gr.Tab("Cohere Command R+"):
211
- cohere_system_message = gr.Textbox(
212
- value="""반드시 한글로 답변할 것.
 
 
213
  너는 최고의 비서이다.
214
  내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.
215
  """,
216
- label="System Message",
217
- lines=3
218
- )
219
- cohere_max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens")
220
- cohere_temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
221
- cohere_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
222
 
223
  cohere_question_input = gr.Textbox(label="질문을 입력하세요")
224
  cohere_answer_output = gr.Textbox(label="답변", interactive=False)
@@ -240,17 +254,19 @@ with gr.Blocks() as demo:
240
 
241
  # --- Tab: ChatGPT ---
242
  with gr.Tab("gpt-4o-mini"):
243
- chatgpt_system_message = gr.Textbox(
244
- value="""반드시 한글로 답변할 것.
 
 
245
  너는 ChatGPT, OpenAI에서 개발한 언어 모델이다.
246
  내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
247
  """,
248
- label="System Message",
249
- lines=3
250
- )
251
- chatgpt_max_tokens = gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max Tokens")
252
- chatgpt_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
253
- chatgpt_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
254
 
255
  chatgpt_question_input = gr.Textbox(label="질문을 입력하세요")
256
  chatgpt_answer_output = gr.Textbox(label="답변", interactive=False)
@@ -318,47 +334,42 @@ with gr.Blocks() as demo:
318
  except Exception as e:
319
  return f"예상치 못한 오류가 발생했습니다: {str(e)}"
320
 
 
321
  with gr.Tab("claude-3-haiku"):
322
  gr.Markdown("claude-3-haiku모델")
323
 
324
- claude_token_box = gr.Textbox(
325
- label="Claude 토큰 (비공개)",
326
- type="password",
327
- placeholder="Claude API 토큰을 입력하세요...",
328
- show_copy_button=False
329
- )
330
 
331
- claude_system_message = gr.Textbox(
332
- label="System Message",
333
- value="""반드시 한글로 답변할 것.
 
 
334
  너는 Anthropic에서 개발한 클로드이다.
335
  최대한 정확하고 친절하게 답변하라.""",
336
- lines=3
337
- )
338
-
339
- claude_max_tokens = gr.Slider(
340
- minimum=1,
341
- maximum=4096,
342
- value=512,
343
- step=1,
344
- label="Max Tokens"
345
- )
346
-
347
- claude_temperature = gr.Slider(
348
- minimum=0.1,
349
- maximum=2.0,
350
- value=0.7,
351
- step=0.05,
352
- label="Temperature"
353
- )
354
-
355
- claude_top_p = gr.Slider(
356
- minimum=0.1,
357
- maximum=1.0,
358
- value=0.95,
359
- step=0.05,
360
- label="Top-p"
361
- )
362
 
363
  claude_question_input = gr.Textbox(
364
  label="질문을 입력하세요",
@@ -392,4 +403,3 @@ with gr.Blocks() as demo:
392
  #############################
393
  if __name__ == "__main__":
394
  demo.launch()
395
-
 
156
  gr.Markdown("# Prompting AI - 일반 문답형 데모")
157
  gr.Markdown("언어모델별 문답형 테스트 데모입니다. 한 번에 한 질문씩만 주고받습니다.")
158
 
159
+ # 요청 사항 1: 토큰 순서 변경
160
  with gr.Row():
161
  hf_token_box = gr.Textbox(
162
  label="HuggingFace 토큰 (비공개)",
 
168
  type="password",
169
  placeholder="OpenAI API 토큰을 입력하세요..."
170
  )
171
+ # Claude 토큰 박스를 상단으로 이동
172
+ with gr.Row():
173
+ claude_token_box = gr.Textbox(
174
+ label="Claude 토큰 (비공개)",
175
+ type="password",
176
+ placeholder="Claude API 토큰을 입력하세요...",
177
+ show_copy_button=False
178
+ )
179
 
180
  # --- Tab: 일반 모델 ---
181
  with gr.Tab("일반 모델"):
 
184
  label="Language Model (HuggingFace)",
185
  value="Zephyr 7B Beta"
186
  )
187
+
188
+ # 요청 사항 2: System Message/Max Tokens/Temperature/Top-p를 아코디언으로 감싸기
189
+ with gr.Accordion("고급 설정 (일반 모델)", open=False):
190
+ max_tokens = gr.Slider(minimum=0, maximum=2000, value=500, step=100, label="Max Tokens")
191
+ temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
192
+ top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
193
+ system_message = gr.Textbox(
194
+ value="""반드시 한글로 답변할 것.
195
  너는 최고의 비서이다.
196
  내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.
197
  """,
198
+ label="System Message",
199
+ lines=3
200
+ )
201
 
202
  question_input = gr.Textbox(label="질문을 입력하세요")
203
  answer_output = gr.Textbox(label="답변", interactive=False)
 
220
 
221
  # --- Tab: Cohere Command R+ ---
222
  with gr.Tab("Cohere Command R+"):
223
+ # 동일하게 아코디언 적용
224
+ with gr.Accordion("고급 설정 (Cohere)", open=False):
225
+ cohere_system_message = gr.Textbox(
226
+ value="""반드시 한글로 답변할 것.
227
  너는 최고의 비서이다.
228
  내가 요구하는것들을 최대한 자세하고 정확하게 답변하라.
229
  """,
230
+ label="System Message",
231
+ lines=3
232
+ )
233
+ cohere_max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens")
234
+ cohere_temperature = gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature")
235
+ cohere_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
236
 
237
  cohere_question_input = gr.Textbox(label="질문을 입력하세요")
238
  cohere_answer_output = gr.Textbox(label="답변", interactive=False)
 
254
 
255
  # --- Tab: ChatGPT ---
256
  with gr.Tab("gpt-4o-mini"):
257
+ # 동일하게 아코디언 적용
258
+ with gr.Accordion("고급 설정 (ChatGPT)", open=False):
259
+ chatgpt_system_message = gr.Textbox(
260
+ value="""반드시 한글로 답변할 것.
261
  너는 ChatGPT, OpenAI에서 개발한 언어 모델이다.
262
  내가 요구하는 것을 최대한 자세하고 정확하게 답변하라.
263
  """,
264
+ label="System Message",
265
+ lines=3
266
+ )
267
+ chatgpt_max_tokens = gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max Tokens")
268
+ chatgpt_temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
269
+ chatgpt_top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-P")
270
 
271
  chatgpt_question_input = gr.Textbox(label="질문을 입력하세요")
272
  chatgpt_answer_output = gr.Textbox(label="답변", interactive=False)
 
334
  except Exception as e:
335
  return f"예상치 못한 오류가 발생했습니다: {str(e)}"
336
 
337
+ # --- Tab: Claude ---
338
  with gr.Tab("claude-3-haiku"):
339
  gr.Markdown("claude-3-haiku모델")
340
 
341
+ # 이미 상단에서 claude_token_box 정의됨
 
 
 
 
 
342
 
343
+ # 동일하게 아코디언 적용
344
+ with gr.Accordion("고급 설정 (Claude)", open=False):
345
+ claude_system_message = gr.Textbox(
346
+ label="System Message",
347
+ value="""반드시 한글로 답변할 것.
348
  너는 Anthropic에서 개발한 클로드이다.
349
  최대한 정확하고 친절하게 답변하라.""",
350
+ lines=3
351
+ )
352
+ claude_max_tokens = gr.Slider(
353
+ minimum=1,
354
+ maximum=4096,
355
+ value=512,
356
+ step=1,
357
+ label="Max Tokens"
358
+ )
359
+ claude_temperature = gr.Slider(
360
+ minimum=0.1,
361
+ maximum=2.0,
362
+ value=0.7,
363
+ step=0.05,
364
+ label="Temperature"
365
+ )
366
+ claude_top_p = gr.Slider(
367
+ minimum=0.1,
368
+ maximum=1.0,
369
+ value=0.95,
370
+ step=0.05,
371
+ label="Top-p"
372
+ )
 
 
 
373
 
374
  claude_question_input = gr.Textbox(
375
  label="질문을 입력하세요",
 
403
  #############################
404
  if __name__ == "__main__":
405
  demo.launch()