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,
|