auhide commited on
Commit
fcda7c5
1 Parent(s): e1c43e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -35
README.md CHANGED
@@ -1,48 +1,26 @@
1
  ---
2
  license: mit
3
- tags:
4
- - generated_from_trainer
5
  model-index:
6
  - name: chef-gpt-base
7
  results: []
 
 
 
8
  ---
9
 
10
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
- should probably proofread and complete it, then remove this comment. -->
12
-
13
  # chef-gpt-base
14
-
15
- This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
16
 
17
  ## Model description
 
 
18
 
19
- More information needed
20
-
21
- ## Intended uses & limitations
22
-
23
- More information needed
24
-
25
- ## Training and evaluation data
26
-
27
- More information needed
28
-
29
- ## Training procedure
30
-
31
- ### Training hyperparameters
32
-
33
- The following hyperparameters were used during training:
34
- - learning_rate: 0.0003
35
- - train_batch_size: 16
36
- - eval_batch_size: 4
37
- - seed: 42
38
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
39
- - lr_scheduler_type: linear
40
- - num_epochs: 20
41
- - mixed_precision_training: Native AMP
42
 
43
- ### Framework versions
44
 
45
- - Transformers 4.26.1
46
- - Pytorch 1.13.0
47
- - Datasets 2.1.0
48
- - Tokenizers 0.13.2
 
1
  ---
2
  license: mit
 
 
3
  model-index:
4
  - name: chef-gpt-base
5
  results: []
6
+ language:
7
+ - bg
8
+ pipeline_tag: text-generation
9
  ---
10
 
 
 
 
11
  # chef-gpt-base
12
+ GPT-2 architecture trained to generate recipes based on ingredients.
 
13
 
14
  ## Model description
15
+ This is GPT-2 pretrained on a custom dataset of recipies in Bulgarian.
16
+ You can find the dataset [here](https://www.kaggle.com/datasets/auhide/bulgarian-recipes-dataset).
17
 
18
+ ## Usage
19
+ ```python
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
+ ```
22
 
23
+ ## Additional tokens
24
+ - [ING] - ingredients token; denotes the begining of the tokens representing the ingredients
25
+ - [EOL] - end-of-line token; equivalent to a newline
26
+ - [REC] - recipe token; denotes the begining of the recipe