iko-01 commited on
Commit
1f23888
·
verified ·
1 Parent(s): d1309e9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -3
README.md CHANGED
@@ -1,3 +1,47 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - ar
5
+ base_model:
6
+ - iko-01/ARABIC_poetry
7
+ pipeline_tag: text-generation
8
+ library_name: transformers
9
+ ---
10
+
11
+
12
+ ```python
13
+ from transformers import pipeline
14
+
15
+ gen = pipeline("text-generation", model="iko-01/ARABIC_poetry2", tokenizer="iko-01/ARABIC_poetry2", device=0 if torch.cuda.is_available() else -1)
16
+
17
+ # ضع هنا أمثلة الاختبار (كلمات مفردة أو جمل قصيرة) — يمكنك تعديلها أو إضافة المزيد
18
+ prompts = [
19
+ "يا لَطيفَ",
20
+ "قَدْ سمعْتُ",
21
+ "حينَ اشتدَّ",
22
+ "يا ذاكرةَ",
23
+ "قَلبي",
24
+ "في الليلِ",
25
+ ]
26
+
27
+ results = []
28
+ for p in prompts:
29
+ out = gen(p,
30
+ max_new_tokens=150,
31
+ do_sample=True,
32
+ top_k=50,
33
+ top_p=0.95,
34
+ temperature=0.8,
35
+ num_return_sequences=1)
36
+ text = out[0]["generated_text"]
37
+ # نريد فقط الجزء الجديد بعد الـ prompt (أو يمكن إخراج السطر كله)
38
+ # هنا نحتفظ بالنص الكامل لمرونة العرض
39
+ results.append({"prompt": p, "output": text})
40
+
41
+ # عرض سريع
42
+ for r in results:
43
+ print("PROMPT:", r["prompt"])
44
+ print("OUTPUT:", r["output"])
45
+ print("-"*40)
46
+
47
+ ```