Update app.py

app.py

@@ -36,7 +36,8 @@ def finetune_small_subset():
       4) Saves the LoRA adapter to "finetuned_myr1".
       5) Reloads the adapter for inference.
     """
-    ds = load_dataset("ServiceNow-AI/R1-Distill-SFT", split="train")
+    # Specify the configuration ("v0" or "v1") explicitly.
+    ds = load_dataset("ServiceNow-AI/R1-Distill-SFT", "v0", split="train")
     ds = ds.select(range(min(NUM_EXAMPLES, len(ds))))
 
     bnb_config = BitsAndBytesConfig(