Views
No views yet
1from transformers import pipeline
2
3gen = pipeline("text-generation", model="iko-01/ARABIC_poetry2", tokenizer="iko-01/ARABIC_poetry2", device=0 if torch.cuda.is_available() else -1)
4
5# ضع هنا أمثلة الاختبار (كلمات مفردة أو جمل قصيرة) — يمكنك تعديلها أو إضافة المزيد
6prompts = [
7 "يا لَطيفَ",
8 "قَدْ سمعْتُ",
9 "حينَ اشتدَّ",
10 "يا ذاكرةَ",
11 "قَلبي",
12 "في الليلِ",
13]
14
15results = []
16for p in prompts:
17 out = gen(p,
18 max_new_tokens=150,
19 do_sample=True,
20 top_k=50,
21 top_p=0.95,
22 temperature=0.8,
23 num_return_sequences=1)
24 text = out[0]["generated_text"]
25 # نريد فقط الجزء الجديد بعد الـ prompt (أو يمكن إخراج السطر كله)
26 # هنا نحتفظ بالنص الكامل لمرونة العرض
27 results.append({"prompt": p, "output": text})
28
29# عرض سريع
30for r in results:
31 print("PROMPT:", r["prompt"])
32 print("OUTPUT:", r["output"])
33 print("-"*40)
34