Create app.py

app.py

@@ -74,8 +74,22 @@ def ensure_loaded():
     if QUESTIONS is not None:
         return
 
-    ds = load_dataset(DATASET_NAME)
-    split = ds["train"]
+   ds = load_dataset(DATASET_NAME)
+
+# ✅ train이 없을 수 있어서 안전하게 split 자동 선택
+if isinstance(ds, dict) and len(ds.keys()) > 0:
+    # DatasetDict일 때
+    if "train" in ds:
+        split = ds["train"]
+    elif "test" in ds:
+        split = ds["test"]
+    elif "validation" in ds:
+        split = ds["validation"]
+    else:
+        split = ds[list(ds.keys())[0]]
+else:
+    # 단일 Dataset일 때
+    split = ds
     out = []
 
     for i,row in enumerate(split):
@@ -251,4 +265,4 @@ with gr.Blocks(css=CSS,title="SAT Math Trainer") as demo:
     nextb.click(next_q, st, [img,weakb,result])
     weakb.click(start_weak, st, [img,weakb,result])
 
-demo.launch()
+demo.launch()