Update README.md

README.md

@@ -4,7 +4,7 @@ license: mit
 
 **Model:** GPT-2
 
-**Model name:** AdsGPT
+**Model name:** AdsGPT2
 
 **Model description:**
 This is a fine-tuned version of the GPT-2 model trained on a dataset of 10,000+ programmatic ad creatives. This model is designed to generate ad content given a product or a brand. For instance, when given the input "Nike Basketball", it will generate a sample ad and also suggest an ad size. The model's main purpose is to inspire ad creatives and provide a starting point for creating effective marketing content.
@@ -24,8 +24,8 @@ You can use this model by providing a product or brand name as an input. For exa
 ```
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
-tokenizer = GPT2Tokenizer.from_pretrained("PeterBrendan/AdsGPT")
-model = GPT2LMHeadModel.from_pretrained("PeterBrendan/AdsGPT")
+tokenizer = GPT2Tokenizer.from_pretrained("PeterBrendan/AdsGPT2")
+model = GPT2LMHeadModel.from_pretrained("PeterBrendan/AdsGPT2")
 
 inputs = tokenizer("Nike Basketball", return_tensors="pt")
 outputs = model.generate(**inputs, max_length=100)