aixsatoshi commited on
Commit
651de2e
1 Parent(s): 76f3d4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -38,7 +38,7 @@ h3 {
38
  .chatbox .messages .message.user {
39
  background-color: #e1f5fe;
40
  }
41
- chatbox .messages .message.bot {
42
  background-color: #eeeeee;
43
  }
44
  """
@@ -59,7 +59,8 @@ print(dataset)
59
  split_name = "train" if "train" in dataset else "test" # デフォルトをtrainにし、なければtestにフォールバック
60
 
61
  # 適切なスプリットから10個の例を取得
62
- examples = random.sample(dataset[split_name], 10)
 
63
  example_inputs = [example['input'] for example in examples]
64
 
65
  @spaces.GPU
 
38
  .chatbox .messages .message.user {
39
  background-color: #e1f5fe;
40
  }
41
+ .chatbox .messages .message.bot {
42
  background-color: #eeeeee;
43
  }
44
  """
 
59
  split_name = "train" if "train" in dataset else "test" # デフォルトをtrainにし、なければtestにフォールバック
60
 
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