Create Main.py

Main.py

@@ -0,0 +1,30 @@
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# تحميل النموذج والتوكنزر
+model_name = "aubmindlab/arabic-gpt2"  # استبدلها باسم النموذج الخاص بك إذا كنت قد أنشأت واحدًا جديدًا
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# إعداد رمز الحشو
+tokenizer.pad_token = "[PAD]"  # تعيين رمز الحشو
+tokenizer.add_special_tokens({'pad_token': '[PAD]'})  # إضافة رمز الحشو
+
+model = AutoModelForCausalLM.from_pretrained(model_name)
+model.resize_token_embeddings(len(tokenizer))
+
+# دالة لتوليد الإجابات
+def generate_response(prompt, max_length=50):
+    input_ids = tokenizer.encode(prompt, return_tensors='pt')
+    
+    # توليد الاستجابة
+    with torch.no_grad():
+        output = model.generate(input_ids, max_length=max_length, pad_token_id=tokenizer.pad_token_id)
+    
+    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    return response
+
+# اختبار النموذج
+if __name__ == "__main__":
+    user_input = "السلام عليكم، كيف يمكنني مساعدتك؟"
+    response = generate_response(user_input)
+    print("الإجابة:", response)