misterkilgore commited on
Commit
960b8bd
1 Parent(s): a18bbca

Completed the mdoel card

Browse files
Files changed (1) hide show
  1. README.md +33 -2
README.md CHANGED
@@ -8,8 +8,39 @@ tags:
8
  widget:
9
  - text: "Il cognitivismo"
10
  - text: "Parliamo di attaccamento"
 
11
  ---
12
 
13
- # DistilGPT2-psy-ita
14
- This model is an Italian-language model finetuned on a corpus of approximately 10000 articles about psychology. Like GPT-2, it can be used to generate italian text about psychological arguments. Users of this model card should also consider information about the design, training, and limitations of DistilGPT2.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
 
 
8
  widget:
9
  - text: "Il cognitivismo"
10
  - text: "Parliamo di attaccamento"
11
+ - text: "Le cause del disturbo d'ansia nei bambini sono"
12
  ---
13
 
14
+ # Italian Psychology DistilGPT-2
15
+
16
+ This model is a fine-tuned version of the GPT-2 language model, trained on a dataset of Italian psychology articles.
17
+ It is capable of generating human-like text on topics related to psychology and mental health in Italian language.
18
+
19
+ ## Model details
20
+
21
+ The base model used for fine-tuning is the GPT-2 model, with a transformer architecture and 1250M parameters.
22
+ The fine-tuning dataset consists of approximately 10,000 Italian psychology articles.
23
+
24
+ ## Example usage
25
+
26
+ ```python
27
+ from transformers import pipeline
28
+
29
+ nlp = pipeline("text-generation", model="misterkilgore/distilgpt2-psy-ita")
30
+
31
+ generated_text = nlp("Le cause del disturbo d'ansia nei bambini sono", max_length=100)
32
+ print(generated_text)
33
+ ```
34
+
35
+ ## Limitations and bias
36
+ This model has been trained on a dataset of Italian psychology articles and may not perform well on other types of text or in other languages.
37
+ Additionally, the dataset used to fine-tune the model may contain biases and limitations, which will be reflected in the generated text.
38
+
39
+ ## Dataset
40
+ The dataset used to fine-tune this model is composed of Italian psychology articles.
41
+ It contains various topics on mental health and psychology, but some limitations and biases may be present. This model is meant to be used only for research and educational purposes.
42
+
43
+ ## Training data
44
+ The training data is composed of Italian psychology articles.
45
+ Fine-tuning was performed on this dataset to adapt the base GPT-2 model to the specific topic of psychology and mental health in Italian language.
46