Update README.md

README.md

@@ -16,7 +16,8 @@ from transformers import pipeline
 model_name = "amitayusht/ProofWala-Multilingual"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-pipeline = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+pipeline = pipeline("text2text-generation", model=model, tokenizer=tokenizer, device=-1) # device=0 for GPU, -1 for CPU
+
 # Example usage
 state = """
 Goals to prove: