Create README.md

README.md

@@ -3,12 +3,43 @@ datasets:
 - truongpdd/vietnamese_poetry
 language:
 - vi
-metrics:
-- accuracy
 base_model:
-- openai-community/gpt2
 - NlpHUST/gpt2-vietnamese
-new_version: openai-community/gpt2
 pipeline_tag: text-generation
-library_name: transformers
----
+---
+To use model, follow the below code:
+```Python
+from transformers import pipeline
+pipe = pipeline("text-generation", model=model.model, tokenizer=model.tokenizer, device=model.device)
+
+# Ví dụ sử dụng pipeline
+prompt_text = "Việt Nam anh hùng"
+print(f"\nPrompt: '{prompt_text}'")
+
+try:
+    generated_output = pipe(
+        prompt_text,
+        max_length=100, # max length of poet
+        num_return_sequences=1, # max poet generated
+        do_sample=True,
+        temperature=0.7, # creative param
+        top_k=30,      # get highest proba of next 30 token
+        top_p=0.85,    # highest proba of token
+        no_repeat_ngram_size=3,
+        repetition_penalty=1.15,
+        pad_token_id=model.tokenizer.pad_token_id,
+        eos_token_id=model.tokenizer.eos_token_id,
+    )
+
+    print(f"\nBài thơ với từ khoá {prompt_text}:")
+    for i, seq in enumerate(generated_output):
+        poem_text = seq['generated_text']
+        print(f"Bài thơ {i+1}:")
+        print(poem_text)
+        print("-" * 30)
+
+except Exception as e:
+    print(f"Lỗi khi sử dụng pipeline: {e}")
+
+print("-------------------------------------")
+```