YoussefAnwar's picture
Update app.py
c8d03cb verified
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()