Update README.md

README.md

@@ -19,8 +19,10 @@ pipeline_tag: text-generation
 ```python
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 import torch
+
 tokenizer = AutoTokenizer.from_pretrained("UpMath/Thai-HomeworkGen-4B-v2")
 model = AutoModelForCausalLM.from_pretrained("UpMath/Thai-HomeworkGen-4B-v2",device_map="auto")
+
 generator = pipeline(
     "text-generation",
     model=model,
@@ -31,6 +33,7 @@ generator = pipeline(
     temperature=0.7,
     pad_token_id=tokenizer.eos_token_id
 )
+
 prompt = """
   โปรดสร้างโจทย์คณิตศาสตร์ใหม่ โดยมีรายละเอียดดังนี้:
   ระดับ Bloom: ["วิเคราะห์"] # หรือ ["วิเคราะห์", "จดจำ", .....]
@@ -38,6 +41,7 @@ prompt = """
   รูปแบบ: ปรนัย # หรือ อัตนัย
   โปรดสร้างโจทย์ พร้อมวิธีทำและคำตอบ:
 """
+
 result = generator(prompt)[0]['generated_text']
 print(result)
 ```