Update README.md

README.md

@@ -62,7 +62,7 @@ are simply reacting to it, which is how markets should operate."""
 import transformers
 import os
 import torch
-os.environ["CUDA_VISIBLE_DEVICES"]="5"
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(device)