Sakalti commited on
Commit
d556900
1 Parent(s): 8d5aaed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -27,6 +27,9 @@ def respond(input_text, system_message, max_tokens, temperature, top_p):
27
  # トークンをテキストにデコード
28
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
29
 
 
 
 
30
  # レスポンスを返す
31
  return response
32
 
@@ -36,7 +39,7 @@ with gr.Blocks() as demo:
36
 
37
  # 追加の入力フィールドをリストで設定
38
  additional_inputs = [
39
- gr.Textbox(value="ユーザーの質問や依頼にのみ答えてください。勝手に会話を進めないでください。", label="システムメッセージ"),
40
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="新規トークン最大"),
41
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度"),
42
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (核サンプリング)")
@@ -45,8 +48,8 @@ with gr.Blocks() as demo:
45
  # ユーザーのメイン入力
46
  input_text = gr.Textbox(label="ユーザー入力", placeholder="質問やテキストを入力してください")
47
 
48
- # 出力エリア("respond"という名前に変更)
49
- output_text = gr.Textbox(label="respond")
50
 
51
  # ボタンとアクション
52
  submit_button = gr.Button("送信")
 
27
  # トークンをテキストにデコード
28
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
29
 
30
+ # ユーザー入力部分を削除
31
+ response = response.replace(input_text_combined, "").strip()
32
+
33
  # レスポンスを返す
34
  return response
35
 
 
39
 
40
  # 追加の入力フィールドをリストで設定
41
  additional_inputs = [
42
+ gr.Textbox(value="ユーザーの質問や依頼にのみ答えてください。ポジティブに答えてください。勝手に会話を進めないでください。", label="システムメッセージ"),
43
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="新規トークン最大"),
44
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="温度"),
45
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (核サンプリング)")
 
48
  # ユーザーのメイン入力
49
  input_text = gr.Textbox(label="ユーザー入力", placeholder="質問やテキストを入力してください")
50
 
51
+ # 出力エリア
52
+ output_text = gr.Textbox(label="応答")
53
 
54
  # ボタンとアクション
55
  submit_button = gr.Button("送信")