trtd56 commited on
Commit
b95d673
1 Parent(s): 6334817

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +4 -3
script.py CHANGED
@@ -1,9 +1,10 @@
1
  import pandas as pd
2
 
3
  print("################################")
4
- from datasets import load_dataset
5
- test_dset = load_dataset("ai-shift/compe_test", data_files={"test": "test.csv"})
6
- print(test_dset)
 
7
  print("################################")
8
 
9
  df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])
 
1
  import pandas as pd
2
 
3
  print("################################")
4
+ from huggingface_hub import hf_hub_download
5
+ hf_hub_download(repo_id="ai-shift/compe_test_data", filename="test.csv", local_dir="./")
6
+
7
+ test_df = pd.read_csv("test.csv")
8
  print("################################")
9
 
10
  df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])