Update app.py

app.py

@@ -13,7 +13,9 @@ max_length = 512
 max_target_length = 128
 
 # Load your dataset
-dataset = datasets.load_dataset("minh21/cpgQA-v1.0-unique-context-test-10-percent", split="test[:10]")
+dataset = datasets.load_dataset("minh21/cpgQA-v1.0-unique-context-test-10-percent", split="test")
+dataset = dataset.shuffle()
+dataset = dataset[:5]
 
 
 # Define your function to generate answers