legacy107 commited on
Commit
5658382
1 Parent(s): 3663dc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ max_target_length = 128
15
  # Load your dataset
16
  dataset = datasets.load_dataset("minh21/cpgQA-v1.0-unique-context-test-10-percent", split="test")
17
  dataset = dataset.shuffle()
18
- dataset = dataset[:5]
19
 
20
 
21
  # Define your function to generate answers
 
15
  # Load your dataset
16
  dataset = datasets.load_dataset("minh21/cpgQA-v1.0-unique-context-test-10-percent", split="test")
17
  dataset = dataset.shuffle()
18
+ dataset = dataset.select(range(5))
19
 
20
 
21
  # Define your function to generate answers