AhsanFarabi
/

python-assistant

Text Generation

Model card Files Files and versions

AhsanFarabi commited on May 9

Commit

4ac403b

·

verified ·

1 Parent(s): f02edcc

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -51,8 +51,8 @@ def is_prime(n):
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("TheAhsanFarabi/python-assistant")
-tokenizer = AutoTokenizer.from_pretrained("TheAhsanFarabi/python-assistant")
 prompt = "Write a function to reverse a string."
 inputs = tokenizer(prompt, return_tensors="pt")

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("AhsanFarabi/python-assistant")
+tokenizer = AutoTokenizer.from_pretrained("AhsanFarabi/python-assistant")
 prompt = "Write a function to reverse a string."
 inputs = tokenizer(prompt, return_tensors="pt")