Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2import torch
3
4# اسم الريبو على Hugging Face
5model_repo = "iko-01/IKOARABICX"
6
7# تحميل الـ Tokenizer من النموذج الأساسي (AraGPT2-base) لأن merges_file مفقود في الريبو الخاص بك
8# هذا يحل الخطأ TypeError المتعلق بـ merges_file
9tokenizer = AutoTokenizer.from_pretrained("aubmindlab/aragpt2-base")
10
11# تحميل النموذج الخاص بك (IKOARABICX)
12model = AutoModelForCausalLM.from_pretrained(model_repo)
13
14# إعداد pipeline لتوليد النص
15generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1) # device=0 لـ GPU، -1 لـ CPU
16
17# مثال على prompt للاختبار (يمكن تغييره)
18prompt = "اشرح مفهوم الذكاء الاصطناعي باختصار."
19
20# توليد النص
21generated_text = generator(
22 prompt,
23 max_length=150, # الحد الأقصى للطول (يشمل الـ prompt)
24 num_return_sequences=1, # عدد الإجابات
25 temperature=0.7, # للتنوع (أقل = أكثر دقة، أعلى = أكثر إبداعًا)
26 top_p=0.9, # لتحسين الجودة
27 do_sample=True, # لتوليد إبداعي
28 pad_token_id=tokenizer.eos_token_id # تجنب تحذيرات
29)
30
31# طباعة النتيجة
32response = generated_text[0]['generated_text'].replace(prompt, "").strip()
33print(f"إجابة النموذج:\n{response}")