ajustes

README.md

@@ -62,12 +62,12 @@ Usage.2:
 $ python3
 >>> from transformers import GPT2LMHeadModel, GPT2Tokenizer, set_seed
 >>> set_seed(42)
->>> model = GPT2LMHeadModel.from_pretrained("egonrp/gpt2-medium-wikiwriter-squadv11-portuguese").to('cuda')
+>>> model = GPT2LMHeadModel.from_pretrained("egonrp/gpt2-medium-wikiwriter-squadv11-portuguese")
 >>> tokenizer = GPT2Tokenizer.from_pretrained("egonrp/gpt2-medium-wikiwriter-squadv11-portuguese")
 >>> tokenizer.add_special_tokens({'pad_token': tokenizer.eos_token})
 >>> model.config.pad_token_id = tokenizer.eos_token_id
 >>> prompt_text = '<|prompter|>A pandemia de COVID-19, também conhecida como pandemia de coronavírus, é uma pandemia em curso de COVID-19, uma doença respiratória aguda causada pelo coronavírus da síndrome respiratória aguda grave 2 (SARS-CoV-2). A doença foi identificada pela primeira vez em Wuhan, na província de Hubei, República Popular da China, em 1 de dezembro de 2019, mas o primeiro caso foi reportado em 31 de dezembro do mesmo ano. Quando foi reportado o primeiro caso de COVID-19?<|assistant|>'
->>> encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt").to('cuda')
+>>> encoded_prompt = tokenizer.encode(prompt_text, add_special_tokens=False, return_tensors="pt")
 >>> output_sequences = model.generate(
     input_ids=encoded_prompt,
     do_sample=False,