iko-01/ARABIC_poetry2

from transformers import pipeline

gen = pipeline("text-generation", model="iko-01/ARABIC_poetry2", tokenizer="iko-01/ARABIC_poetry2", device=0 if torch.cuda.is_available() else -1)

# ضع هنا أمثلة الاختبار (كلمات مفردة أو جمل قصيرة) — يمكنك تعديلها أو إضافة المزيد
prompts = [
    "يا لَطيفَ",
    "قَدْ سمعْتُ",
    "حينَ اشتدَّ",
    "يا ذاكرةَ",
    "قَلبي",
    "في الليلِ",
]

results = []
for p in prompts:
    out = gen(p,
              max_new_tokens=150,
              do_sample=True,
              top_k=50,
              top_p=0.95,
              temperature=0.8,
              num_return_sequences=1)
    text = out[0]["generated_text"]
    # نريد فقط الجزء الجديد بعد الـ prompt (أو يمكن إخراج السطر كله)
    # هنا نحتفظ بالنص الكامل لمرونة العرض
    results.append({"prompt": p, "output": text})

# عرض سريع
for r in results:
    print("PROMPT:", r["prompt"])
    print("OUTPUT:", r["output"])
    print("-"*40)