Update README.md

README.md

@@ -66,9 +66,13 @@ tokenizer = AutoTokenizer.from_pretrained("LingoIITGN/ganga-1b")
 model = AutoModelForCausalLM.from_pretrained("LingoIITGN/ganga-1b", device_map="auto")
 
 input_text = "BCCI ने टी-20 वर्ल्ड कप के बीच जिम्बाब्वे सीरीज "
-input_ids = tokenizer.encode(input_text, return_tensors="pt").to("cuda")
+input_ids = tokenizer.encode(input_text,
+            return_tensors="pt").to("cuda")
+
+outputs = model.generate(input_ids, max_new_tokens=100,
+          do_sample=True, top_k=50,
+          top_p=0.95, temperature=0.7)
 
-outputs = model.generate(input_ids, max_new_tokens=100, do_sample=True, top_k=50, top_p=0.95, temperature=0.7)
 print(tokenizer.decode(outputs[0]))
 
 ```