Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| pipe = pipeline("text2text-generation", model="YoussefAnwar/mt5-summarize-ar-en") | |
| gen_config = { | |
| 'max_length': 128, | |
| 'num_beams': 15, | |
| 'length_penalty': 0.6, | |
| 'no_repeat_ngram_size': 2 | |
| } | |
| def summarize(text): | |
| summary = pipe(text, **gen_config) | |
| return summary[0]['generated_text'] | |
| interface = gr.Interface( | |
| fn=summarize, | |
| inputs=gr.Textbox(lines=10, placeholder="Enter text to summarize..."), | |
| outputs="text", | |
| title="Arabic-English Text Summarizer", | |
| description="Enter Arabic or English text, and the model will generate a summary." | |
| ) | |
| interface.launch() | |