PeterBrendan
commited on
Commit
·
9a5ef44
1
Parent(s):
1b583b3
Update README.md
Browse files
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
|
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:**
|