PeterBrendan commited on
Commit
9a5ef44
·
1 Parent(s): 1b583b3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -24,8 +24,8 @@ You can use this model by providing a product or brand name as an input. For exa
24
  ```
25
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
26
 
27
- tokenizer = GPT2Tokenizer.from_pretrained("AdsGPT")
28
- model = GPT2LMHeadModel.from_pretrained("AdsGPT")
29
 
30
  inputs = tokenizer("Nike Basketball", return_tensors="pt")
31
  outputs = model.generate(**inputs, max_length=100)
@@ -38,7 +38,7 @@ This model was trained on a dataset consisting of over 10,000 programmatic ad cr
38
 
39
 
40
  **Training procedure:**
41
- The model was fine-tuned using the GPT-2 base model with the aforementioned training data. The average training loss was 0.16038204793556174.
42
 
43
 
44
  **Evaluation results:**
 
24
  ```
25
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
26
 
27
+ tokenizer = GPT2Tokenizer.from_pretrained("PeterBrendan/AdsGPT")
28
+ model = GPT2LMHeadModel.from_pretrained("PeterBrendan/AdsGPT")
29
 
30
  inputs = tokenizer("Nike Basketball", return_tensors="pt")
31
  outputs = model.generate(**inputs, max_length=100)
 
38
 
39
 
40
  **Training procedure:**
41
+ The model was fine-tuned using the GPT-2 base model with the aforementioned training data. The training loss was 0.16038204793556174.
42
 
43
 
44
  **Evaluation results:**