Tutorial / app.py
eminAydin's picture
Update app.py
457f99e verified
raw
history blame
No virus
1.36 kB
from transformers import GPT2LMHeadModel, GPT2Tokenizer
import gradio as gr
# Model ve tokenizer'ı yükle
model_name = "eminAydin/turkish-gpt2-mini-M1-cleaned-sports720k-10ep"
tokenizer = GPT2Tokenizer.from_pretrained("ytu-ce-cosmos/turkish-gpt2")
tokenizer.pad_token = tokenizer.eos_token
model = GPT2LMHeadModel.from_pretrained(model_name)
# parametrelere bağlı olarak text üretmek
generation_config = {
'temperature': 0.7, # rastgelelik için artırabilirim
"do_sample": True,
"max_new_tokens": 50,
'top_p': 0.9, # Use top-p sampling
'repetition_penalty': 1.3, # tekrarı azaltır fakat anlamı bozabilir.
'eos_token_id': tokenizer.eos_token_id, # Set eos_token_id
'pad_token_id': tokenizer.pad_token_id, # Set pad_token_id
}
def launch(input):
input_ids = tokenizer.encode(input, return_tensors="pt")
output = model.generate(input_ids, num_return_sequences=1, **generation_config)
output = tokenizer.decode(output[0], skip_special_tokens=True)
return output
iface = gr.Interface(launch,
inputs="text",
outputs="text",
title="Turkish Text Generation with GPT-2",
description="Enter a Turkish prompt and generate text using GPT-2.",
theme="default")
iface.launch()