Spaces:
Running
on
Zero
Running
on
Zero
aixsatoshi
commited on
Commit
•
0699d9c
1
Parent(s):
e62d5ad
Update app.py
Browse files
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,
|