mehmet0789's picture
Upload 2 files
47658f6 verified
raw
history blame
1.18 kB
import gradio as gr
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqGeneration
# Model ve tokenizer'ı yükle
model_name = "ozcangundes/mt5-small-turkish-summarization"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name)
summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
def summarize_text(text):
if len(text.split()) < 100:
return "Metin çok kısa. En az 100 kelime olmalıdır."
# Metni özetle
summary = summarizer(text,
max_length=150,
min_length=50,
length_penalty=2.0,
num_beams=4,
early_stopping=True)
return summary[0]['summary_text']
# Gradio arayüzünü oluştur
iface = gr.Interface(
fn=summarize_text,
inputs=gr.Textbox(lines=10, label="Özetlenecek Metin"),
outputs=gr.Textbox(label="Özet"),
title="Türkçe Metin Özetleme",
description="Türkçe metinleri özetleyen yapay zeka modeli. En az 100 kelimelik metin giriniz."
)
iface.launch()