Update app.py

app.py

@@ -32,10 +32,10 @@ def fine_tune_model(base_model_name, dataset_name):
 
     print("### Dataset")
     print(dataset)
-    print("###")
-    print(dataset["train"][:5])
-    print("###")
-    print(dataset["test"][:5])
+    print("### Train example")
+    print(dataset["train"][:1])
+    print("### Test example")
+    print(dataset["test"][:1)
     print("###")
     
     # Load model
@@ -57,15 +57,17 @@ def fine_tune_model(base_model_name, dataset_name):
 
     print("### Pre-processed dataset")
     print(dataset)
-    print("###")
-    print(dataset["train"][:5])
-    print("###")
-    print(dataset["test"][:5])
+    print("### Train example")
+    print(dataset["train"][:1])
+    print("### Test example")
+    print(dataset["test"][:1)
     print("###")
     
     # Split dataset into training and validation sets
-    train_dataset = dataset["train"] #.shuffle(seed=42).select(range(1000))
-    test_dataset = dataset["test"] #.shuffle(seed=42).select(range(100))
+    #train_dataset = dataset["train"]
+    #test_dataset = dataset["test"]
+    train_dataset = dataset["train"].shuffle(seed=42).select(range(100))
+    test_dataset = dataset["test"].shuffle(seed=42).select(range(10))
 
     print("### Training dataset")
     print(train_dataset)