Upload 2 files

app.py

@@ -4,28 +4,28 @@ from transformers import pipeline
 # تحميل نموذج التلخيص (facebook/bart-large-cnn)
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# تحميل نموذج كتابة المقالات (gpt2)
-generator = pipeline("text-generation", model="gpt2")
+# تحميل نموذج كتابة المقالات بالعربي (aragpt2-base)
+generator = pipeline("text-generation", model="aubmindlab/aragpt2-base")
 
-# دالة لتلخيص النص
+# دالة لتلخيص النص (بالإنجليزي حاليًا لأن النموذج إنجليزي)
 def summarize_text(text):
     summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
     return summary[0]["summary_text"]
 
-# دالة لكتابة مقال
+# دالة لكتابة مقال بالعربي
 def generate_article(prompt):
-    article = generator(prompt, max_length=200, num_return_sequences=1)
+    article = generator(prompt, max_length=200, num_return_sequences=1, do_sample=True)
     return article[0]["generated_text"]
 
 # واجهة Gradio
-with gr.Blocks(title="أداة كتابة المقالات وتلخيصها القبطان تك") as interface:
+with gr.Blocks(title="أداة كتابة المقالات وتلخيصها") as interface:
     gr.Markdown("## أداة كتابة المقالات وتلخيصها")
     with gr.Tab("تلخيص النصوص"):
-        input_text = gr.Textbox(label="أدخل النص للتلخيص", lines=5)
+        input_text = gr.Textbox(label="أدخل النص للتلخيص (بالإنجليزي حاليًا)", lines=5)
         output_text = gr.Textbox(label="الملخص", lines=3)
         gr.Button("تلخيص").click(summarize_text, inputs=input_text, outputs=output_text)
     with gr.Tab("كتابة مقال"):
-        input_prompt = gr.Textbox(label="أدخل بداية المقال", lines=2)
+        input_prompt = gr.Textbox(label="أدخل بداية المقال (بالعربي)", lines=2)
         output_article = gr.Textbox(label="المقال المكتوب", lines=5)
         gr.Button("اكتب مقال").click(generate_article, inputs=input_prompt, outputs=output_article)