Spaces:
Paused
Paused
| import gradio as gr | |
| from transformers import AutoModelForSeq2SeqLM, AutoTokenizer | |
| import torch | |
| # تحميل النموذج والتوكينايزر | |
| model_name = "methodya/arabic-summarizer-philosophy-v2" | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| model = AutoModelForSeq2SeqLM.from_pretrained(model_name) | |
| # التأكد من استخدام GPU إذا كان متوفراً | |
| device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') | |
| model = model.to(device) | |
| model.eval() | |
| def summarize(text): | |
| # التأكد من أن النص ليس فارغاً | |
| if not text.strip(): | |
| return "الرجاء إدخال نص للتلخيص" | |
| # تحضير المدخلات | |
| inputs = tokenizer(text, return_tensors="pt", max_length=2048, padding=True, truncation=True) | |
| inputs = {k: v.to(device) for k, v in inputs.items()} | |
| # توليد الملخص | |
| with torch.no_grad(): | |
| outputs = model.generate( | |
| **inputs, | |
| max_length=512, | |
| min_length=30, | |
| num_beams=4, | |
| length_penalty=2.0, | |
| no_repeat_ngram_size=3 | |
| ) | |
| # فك ترميز الملخص | |
| summary = tokenizer.decode(outputs[0], skip_special_tokens=True) | |
| return summary | |
| # تعريف واجهة gradio | |
| iface = gr.Interface( | |
| fn=summarize, | |
| inputs=gr.Textbox(label="النص", placeholder="أدخل النص العربي هنا..."), | |
| outputs=gr.Textbox(label="الملخص"), | |
| title="نظام تلخيص النصوص الفلسفية العربية", | |
| description="نموذج لتلخيص النصوص الفلسفية باللغة العربية", | |
| submit_btn="تلخيص", | |
| clear_btn="مسح" | |
| ) | |
| # تشغيل التطبيق | |
| iface.launch() |