josu commited on
Commit
ab20f95
1 Parent(s): 4964527

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -12,8 +12,8 @@ widget:
12
  from transformers import GenerationConfig
13
  from transformers import AutoModelForCausalLM, AutoTokenizer
14
 
15
- model = AutoModelForCausalLM.from_pretrained("josu/gpt-neo-pt-br")
16
- tokenizer = AutoTokenizer.from_pretrained("josu/gpt-neo-pt-br")
17
 
18
 
19
  def generate_prompt(instruction, input=None):
 
12
  from transformers import GenerationConfig
13
  from transformers import AutoModelForCausalLM, AutoTokenizer
14
 
15
+ model = AutoModelForCausalLM.from_pretrained("josu/gpt-neo-br-instruction")
16
+ tokenizer = AutoTokenizer.from_pretrained("josu/gpt-neo-br-instruction")
17
 
18
 
19
  def generate_prompt(instruction, input=None):