Update README.md

README.md

@@ -34,11 +34,36 @@ This model is an attempt to simplify code understanding by generating line by li
 # Model Usage
 
 ```py
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import (
+    AutoModelForSeq2SeqLM,
+    AutoTokenizer,
+    AutoConfig,
+    pipeline,
+)
 
-tokenizer = AutoTokenizer.from_pretrained("sagard21/python-code-explainer")
+model_name = "sagard21/python-code-explainer"
 
-model = AutoModelForSeq2SeqLM.from_pretrained("sagard21/python-code-explainer")
+tokenizer = AutoTokenizer.from_pretrained(model_name, padding=True)
+
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+config = AutoConfig.from_pretrained(model_name)
+
+model.eval()
+
+pipe = pipeline("summarization", model=model_name, config=config, tokenizer=tokenizer)
+
+raw_code = """
+def preprocess(text: str) -> str:
+    text = str(text)
+    text = text.replace("\n", " ")
+    tokenized_text = text.split(" ")
+    preprocessed_text = " ".join([token for token in tokenized_text if token])
+
+    return preprocessed_text
+"""
+
+print(pipe(raw_code)[0]["summary_text"])
 
 ```