Update README.md

README.md

@@ -26,6 +26,9 @@ This model is based on [Telugu-Llama2-7B-v0-Base](https://huggingface.co/Telugu-
 ```
 
 # Usage
+
+## With Romanized Telugu
+
 ```python3
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
@@ -37,33 +40,59 @@ model_name = "Telugu-LLM-Labs/Telugu-Llama2-7B-v0-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="right")
 model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
 
+instruction = "Krindi samaacharam prakaram google app eppudu release ayyindi?"
+input ="Google News is a news aggregator service developed by Google. It presents a continuous flow of links to articles organized from thousands of publishers and magazines. Google News is available as an app on Android, iOS, and the Web. Google released a beta version in September 2002 and the official app in January 2006."
+
 text = f"""Instruction: {instruction} \nInput: {input} \nResponse:"""
 
 encodings = tokenizer(text, padding=True, return_tensors="pt")
 encodings = encodings.to(device)
 
 with torch.inference_mode():
-    outputs = model.generate(encodings.input_ids, do_sample=False, max_new_tokens=100)
+    outputs = model.generate(encodings.input_ids, do_sample=False, max_new_tokens=500)
 
 output = tokenizer.batch_decode(outputs.detach(), skip_special_tokens=True)
 ```
 
-## Examples: 
-
-### With Romanized Telugu
+### Sample Output:
 
 ```
-instruction = "Krindi samaacharam prakaram google app eppudu release ayyindi?"
-input ="Google News is a news aggregator service developed by Google. It presents a continuous flow of links to articles organized from thousands of publishers and magazines. Google News is available as an app on Android, iOS, and the Web. Google released a beta version in September 2002 and the official app in January 2006."
+1. September 2002 Google released a beta version of Google News.
+2. January 2006 Google released the official version of Google News.
 ```
 
-### With Native Telugu
+## With Native Telugu
+
+```python3
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+model_name = "Telugu-LLM-Labs/Telugu-Llama2-7B-v0-Instruct"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="right")
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
 
-```
 instruction = "కింది వచనాన్ని సంగ్రహించండి"
 input="గూగుల్ వార్తలు అనేది గూగుల్ ద్వారా అభివృద్ధి చేయబడిన వార్తా అగ్రిగేటర్ సేవ. ఇది వేలకొద్దీ ప్రచురణకర్తలు మరియు మ్యాగజైన్‌ల నుండి నిర్వహించబడిన కథనాలకు నిరంతర లింక్‌లను అందిస్తుంది. గూగుల్ వార్తలు Android, iOS మరియు వెబ్‌లో యాప్‌గా అందుబాటులో ఉన్నాయి. గూగుల్ సెప్టెంబరు 2002లో బీటా వెర్షన్‌ను మరియు జనవరి 2006లో అధికారిక యాప్‌ను విడుదల చేసింది."
+
+text = f"""Instruction: {instruction} \nInput: {input} \nResponse:"""
+
+encodings = tokenizer(text, padding=True, return_tensors="pt")
+encodings = encodings.to(device)
+
+with torch.inference_mode():
+    outputs = model.generate(encodings.input_ids, do_sample=False, max_new_tokens=500)
+
+output = tokenizer.batch_decode(outputs.detach(), skip_special_tokens=True)
 ```
 
+### Sample Output:
+
+1. గూగుల్ వార్తలు అనేది గూగుల్ ద్వారా అభివృద్ధి చేయబడిన వార్తా అగ్రిగేటర్ సేవ, వేలకొద్దీ ప్రచురణకర్తలు మరియు మ్యాగజైన్‌ల నుండి నిర్వహించబడిన కథనాలకు నిరంతర లింక్‌లను అందిస్తుంది.
+2. గూగుల్ సెప్టెంబరు 2002లో బీటా వెర్షన్ మరియు జనవరి 2006లో అధికారిక యాప్ ను విడుదల చేసింది.
+
 # Developers:
 
 The model is a collaborative effort by [Ravi Theja](https://twitter.com/ravithejads) and [Ramsri Goutham](https://twitter.com/ramsri_goutham). Feel free to DM either of us if you have any questions.