Update README.md

README.md

@@ -41,7 +41,9 @@ def process_data_sample(messages):
         processed_example += f"<|"+role+"|>\n "+content+"\n"
 
     return processed_example
-
+```
+Inference can then be performed as usual with HF models as follows:
+```python
 messages = [
     {"role": "system", "content": "You are a Khaadi Social Media Post Generator who helps with user queries or generate him khaadi posts give only three hashtags and be concise as possible dont try to make up."},
     {"role": "user", "content": "Generate post on new arrival of winter"},
@@ -54,17 +56,6 @@ outputs = model.generate(**inputs, generation_config=generation_config)
 asnwer = tokenizer.decode(outputs[0], skip_special_tokens=True)
 print(asnwer)
 ```
-Inference can then be performed as usual with HF models as follows:
-```python
-prompt = "پیٹرول کی قیمت میں 2روپے 50 پیسے اضافہ"
-formatted_prompt = (
-    f"اس دی گی ایک خبر سے متعلق ایک مضمون لکھیں"
-    f"### Human: {prompt} ### Assistant:"
-)
-inputs = tokenizer(formatted_prompt, return_tensors="pt").to("cuda:0")
-outputs = model.generate(inputs=inputs.input_ids, max_new_tokens=20)
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
-```
 Expected output similar to the following:
 ```