sagard21
/

python-code-explainer

text2text-generation

Trained with AutoTrain

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sagard21 commited on Jan 6, 2023

Commit

67c58d7

·

1 Parent(s): c24b3be

Update README.md

Files changed (1) hide show

README.md +0 -2

README.md CHANGED Viewed

@@ -29,8 +29,6 @@ This model is an attempt to simplify code understanding by generating line by li
 ```py
 from transformers import AutoTokenizer, T5ForConditionalGeneration, SummarizationPipeline
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 pipeline = SummarizationPipeline(
     model=T5ForConditionalGeneration.from_pretrained("sagard21/python-code-explainer"),
     tokenizer=AutoTokenizer.from_pretrained("sagard21/python-code-explainer", skip_special_tokens=True),

 ```py
 from transformers import AutoTokenizer, T5ForConditionalGeneration, SummarizationPipeline
 pipeline = SummarizationPipeline(
     model=T5ForConditionalGeneration.from_pretrained("sagard21/python-code-explainer"),
     tokenizer=AutoTokenizer.from_pretrained("sagard21/python-code-explainer", skip_special_tokens=True),