Update README.md

README.md

@@ -5,6 +5,8 @@ tags:
 model-index:
 - name: flan-t5-small-distil-v2
   results: []
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -41,7 +43,50 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
-### Training results
+## Use
+
+### CPU
+
+<details>
+<summary> Click to expand </summary>
+
+```python
+# pip install -q transformers
+from transformers import pipeline
+
+checkpoint = "{model_name}"
+
+model = pipeline('text2text-generation', model=checkpoint, use_auth_token=True)
+
+input_prompt = 'Please let me know your thoughts on the given place and why you think it deserves to be visited: \n"Barcelona, Spain"'
+generated_text = generator(input_prompt, max_length=512, do_sample=True, repetition_penalty=1.5)[0]['generated_text']
+
+print("Response": generated_text)
+```
+
+</details>
+
+### GPU
+
+<details>
+<summary> Click to expand </summary>
+
+```python
+# pip install -q transformers
+from transformers import pipeline
+
+checkpoint = "{model_name}"
+
+model = pipeline('text2text-generation', model=checkpoint, use_auth_token=True, device=0)
+
+input_prompt = 'Please let me know your thoughts on the given place and why you think it deserves to be visited: \n"Barcelona, Spain"'
+generated_text = generator(input_prompt, max_length=512, do_sample=True, repetition_penalty=1.5)[0]['generated_text']
+
+print("Response": generated_text)
+```
+
+</details>
+
 
 
 
@@ -50,4 +95,4 @@ The following hyperparameters were used during training:
 - Transformers 4.27.0
 - Pytorch 2.0.0+cu117
 - Datasets 2.2.0
-- Tokenizers 0.13.2
+- Tokenizers 0.13.2