Update README.md

README.md

@@ -1,9 +1,24 @@
 Para un uso sencillo del modelo utilize el siguiente codigo:
 
+```py
+ #! pip install transformers
+ #! pip install torch
+ #! pip install datasets
+```
+
 ```py
 from transformers import DistilBertForSequenceClassification, DistilBertTokenizer
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from datasets import load_dataset
 import numpy as np
+import torch
+
+
+dataset = load_dataset("manoh2f2/songs_resampled")
+
+# Cargar el dataset en un DataFrame
+split_name = 'train'
+df_resampled = dataset[split_name].to_pandas()
 
 tokenizer = AutoTokenizer.from_pretrained("manoh2f2/recommend_songs")
 model = AutoModelForSequenceClassification.from_pretrained("manoh2f2/recommend_songs")