TK17250 commited on
Commit
5ed0fd6
·
verified ·
1 Parent(s): d2279d0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -0
README.md CHANGED
@@ -17,6 +17,38 @@ licence: license
17
 
18
  ---
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  ## 🎯 จุดประสงค์ของโมเดล
21
 
22
  - ✅ สร้างโจทย์คณิตศาสตร์ใหม่ในรูปแบบภาษาไทย
 
17
 
18
  ---
19
 
20
+ ## 🧪 ตัวอย่างการใช้งาน
21
+
22
+ ```python
23
+ from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
24
+ import torch
25
+
26
+ tokenizer = AutoTokenizer.from_pretrained("UpMath/Thai-HomeworkGen-7.2B-v1")
27
+ model = AutoModelForCausalLM.from_pretrained("UpMath/Thai-HomeworkGen-7.2B-v1",device_map="auto")
28
+
29
+ generator = pipeline(
30
+ "text-generation",
31
+ model=model,
32
+ tokenizer=tokenizer,
33
+ device_map="auto",
34
+ max_new_tokens=2048,
35
+ do_sample=True,
36
+ temperature=0.7,
37
+ pad_token_id=tokenizer.eos_token_id
38
+ )
39
+
40
+ prompt = """
41
+ ระดับ Bloom: "วิเคราะห์" # หรือ ["วิเคราะห์", "จดจำ"]
42
+ ระดับชั้น: 'มัธยมศึกษาปีที่ 4'
43
+ โปรดสร้างโจทย์ พร้อมวิธีทำและคำตอบ:
44
+ """
45
+
46
+ result = generator(prompt)[0]['generated_text']
47
+ print(result)
48
+ ```
49
+
50
+ ---
51
+
52
  ## 🎯 จุดประสงค์ของโมเดล
53
 
54
  - ✅ สร้างโจทย์คณิตศาสตร์ใหม่ในรูปแบบภาษาไทย