aixsatoshi commited on
Commit
82b85f5
·
verified ·
1 Parent(s): 651de2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,7 +61,7 @@ split_name = "train" if "train" in dataset else "test" # デフォルトをtrai
61
  # 適切なスプリットから10個の例を取得
62
  examples_list = list(dataset[split_name]) # スプリットをリストに変換
63
  examples = random.sample(examples_list, 10) # リストからランダムに10個選択
64
- example_inputs = [example['input'] for example in examples]
65
 
66
  @spaces.GPU
67
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
@@ -153,7 +153,7 @@ with gr.Blocks(css=CSS) as demo:
153
  render=False,
154
  ),
155
  ],
156
- examples=example_inputs,
157
  cache_examples=False,
158
  )
159
 
 
61
  # 適切なスプリットから10個の例を取得
62
  examples_list = list(dataset[split_name]) # スプリットをリストに変換
63
  examples = random.sample(examples_list, 10) # リストからランダムに10個選択
64
+ example_inputs = [[example['input']] for example in examples] # ネストされたリストに変換
65
 
66
  @spaces.GPU
67
  def stream_chat(message: str, history: list, temperature: float, max_new_tokens: int, top_p: float, top_k: int, penalty: float):
 
153
  render=False,
154
  ),
155
  ],
156
+ examples=example_inputs, # ネストされたリストを渡す
157
  cache_examples=False,
158
  )
159