aixsatoshi commited on
Commit
0699d9c
1 Parent(s): e62d5ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -50,6 +50,19 @@ hf_hub_download(
50
  llm = None
51
  llm_model = None
52
 
 
 
 
 
 
 
 
 
 
 
 
 
 
53
  @spaces.GPU(duration=120)
54
  def respond(
55
  message,
 
50
  llm = None
51
  llm_model = None
52
 
53
+ # データセットをロードしてスプリットを確認
54
+ dataset = load_dataset("elyza/ELYZA-tasks-100")
55
+ print(dataset)
56
+
57
+ # 使用するスプリット名を確認
58
+ split_name = "train" if "train" in dataset else "test" # デフォルトをtrainにし、なければtestにフォールバック
59
+
60
+ # 適切なスプリットから10個の例を取得
61
+ examples_list = list(dataset[split_name]) # スプリットをリストに変換
62
+ examples = random.sample(examples_list, 10) # リストからランダムに10個選択
63
+ example_inputs = [[example['input']] for example in examples] # ネストされたリストに変換
64
+
65
+
66
  @spaces.GPU(duration=120)
67
  def respond(
68
  message,