khulnasoft
/

aifixcode-model

code-generation

text2text-generation

code-correction

Model card Files Files and versions

khulnasoft commited on May 14

Commit

384261b

·

verified ·

1 Parent(s): 80f7eae

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,8 +32,8 @@ Use the transformers pipeline or run via CLI:
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-model = AutoModelForSeq2SeqLM.from_pretrained("YOUR_USERNAME/aifixcode-model")
-tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/aifixcode-model")
 input_code = "def foo(x):\n print(x"
 inputs = tokenizer(input_code, return_tensors="pt")

 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+model = AutoModelForSeq2SeqLM.from_pretrained("khulnasoft/aifixcode-model")
+tokenizer = AutoTokenizer.from_pretrained("khulnasoft/aifixcode-model")
 input_code = "def foo(x):\n print(x"
 inputs = tokenizer(input_code, return_tensors="pt")