rwayz commited on
Commit
72f6356
·
verified ·
1 Parent(s): 51b6b2a

Create test.py

Browse files
Files changed (1) hide show
  1. test.py +18 -0
test.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from datasets import load_dataset
3
+ from huggingface_hub import login
4
+
5
+ # 🔒 Obter o Token do ambiente
6
+ HF_API_KEY = os.getenv("HF_API_KEY")
7
+ if not HF_API_KEY:
8
+ raise ValueError("❌ ERRO: Token Hugging Face não encontrado no ambiente.")
9
+
10
+ # ✅ Fazer login no Hugging Face
11
+ login(HF_API_KEY)
12
+
13
+ # ✅ Carregar o dataset
14
+ dataset = load_dataset("rwayz/tributario", token=HF_API_KEY)
15
+
16
+ # ✅ Listar os splits disponíveis
17
+ print("✅ Splits disponíveis no dataset:")
18
+ print(dataset)