Fathi7ma commited on
Commit
38c20fb
·
verified ·
1 Parent(s): d1ec9f2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -20
README.md CHANGED
@@ -1,36 +1,47 @@
1
  ---
2
- library_name: transformers
3
- license: apache-2.0
4
- base_model: distilgpt2
5
  tags:
6
- - generated_from_trainer
 
 
 
 
 
 
 
 
7
  model-index:
8
  - name: tiny-stories-gpt2
9
  results: []
10
  ---
11
 
12
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
- should probably proofread and complete it, then remove this comment. -->
 
 
14
 
15
- # tiny-stories-gpt2
16
-
17
- This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on an unknown dataset.
18
- It achieves the following results on the evaluation set:
19
- - Loss: 1.6995
20
-
21
- ## Model description
22
-
23
- More information needed
24
 
25
- ## Intended uses & limitations
 
 
26
 
27
- More information needed
28
 
29
- ## Training and evaluation data
 
 
 
 
30
 
31
- More information needed
 
 
 
 
 
32
 
33
- ## Training procedure
34
 
35
  ### Training hyperparameters
36
 
 
1
  ---
2
+ language:
3
+ - en
 
4
  tags:
5
+ - text-generation
6
+ - gpt2
7
+ - tiny-stories
8
+ - creative-writing
9
+ license: mit
10
+ datasets:
11
+ - roneneldan/TinyStories
12
+ metrics:
13
+ - perplexity
14
  model-index:
15
  - name: tiny-stories-gpt2
16
  results: []
17
  ---
18
 
19
+ # 🧚‍♀️ Tiny Stories GPT-2
20
+ **Author:** [Fathi7ma](https://huggingface.co/Fathi7ma)
21
+ **Model type:** Fine-tuned GPT-2 for creative story generation
22
+ **Dataset:** [roneneldan/TinyStories](https://huggingface.co/datasets/roneneldan/TinyStories)
23
 
24
+ ---
 
 
 
 
 
 
 
 
25
 
26
+ ## 🧠 Model Overview
27
+ This model is a fine-tuned version of **GPT-2 (small)** trained on the [TinyStories](https://huggingface.co/datasets/roneneldan/TinyStories) dataset — a large collection of short, simple stories for children.
28
+ The goal of this project is to create a lightweight, fun, and creative **story generator** that can produce short, imaginative stories from user prompts.
29
 
30
+ ---
31
 
32
+ ## Intended Use
33
+ Use this model for:
34
+ - Generating **short, creative stories**
35
+ - Educational or entertainment purposes
36
+ - Quick writing inspiration or kids’ content
37
 
38
+ Example use:
39
+ ```python
40
+ from transformers import pipeline
41
+ generator = pipeline("text-generation", model="Fathi7ma/tiny-stories-gpt2")
42
+ prompt = "A tiny robot who wanted to fly"
43
+ print(generator(prompt, max_length=80, num_return_sequences=1)[0]["generated_text"])
44
 
 
45
 
46
  ### Training hyperparameters
47