Update README.md

README.md

@@ -44,8 +44,8 @@ pipeline_tag: text2text-generation
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
 ### Direct Use
-
-`from peft import PeftModel
+```python
+from peft import PeftModel
 
 model_id = 'google-t5/t5-base'
 
@@ -58,7 +58,6 @@ bnb_config = BitsAndBytesConfig(
 
 original_model = AutoModelForSeq2SeqLM.from_pretrained(model_id,quantization_config=bnb_config,device_map='auto')
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-
 tokenizer.pad_token = tokenizer.eos_token
 
 peft_model = PeftModel.from_pretrained(original_model, "bhuvanmdev/t5-base-news-describer")
@@ -74,15 +73,16 @@ generation_config.eos_token_id = tokenizer.eos_token_id
 generation_config.use_cache = True
 
 prompt = f"""Title: A big accidient occurs in luxemberg.""".strip()
-
 encoding = tokenizer(prompt, return_tensors="pt").to(device)
+
 with torch.inference_mode():
     outputs = peft_model.generate(
         input_ids=encoding.input_ids,
         attention_mask=encoding.attention_mask,
         generation_config=generation_config,
     )
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))`
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
 
 [More Information Needed]