khulnasoft commited on
Commit
384261b
·
verified ·
1 Parent(s): 80f7eae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -32,8 +32,8 @@ Use the transformers pipeline or run via CLI:
32
 
33
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
34
 
35
- model = AutoModelForSeq2SeqLM.from_pretrained("YOUR_USERNAME/aifixcode-model")
36
- tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/aifixcode-model")
37
 
38
  input_code = "def foo(x):\n print(x"
39
  inputs = tokenizer(input_code, return_tensors="pt")
 
32
 
33
  from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
34
 
35
+ model = AutoModelForSeq2SeqLM.from_pretrained("khulnasoft/aifixcode-model")
36
+ tokenizer = AutoTokenizer.from_pretrained("khulnasoft/aifixcode-model")
37
 
38
  input_code = "def foo(x):\n print(x"
39
  inputs = tokenizer(input_code, return_tensors="pt")