yonkasoft's picture
Upload 2 files
7044ef9 verified
raw
history blame
No virus
681 Bytes
# pip install accelerate
from transformers import T5Tokenizer, T5ForConditionalGeneration
tokenizer = T5Tokenizer.from_pretrained("dbmdz/bert-base-turkish-cased")
model = T5ForConditionalGeneration.from_pretrained("dbmdz/bert-base-turkish-cased", device_map="auto")
input_text = "Başlık: Dijital Pazarlama, Alt başlıklar: dijital pazarlama nasıl yapılır?, dijital pazarlama ve gelişimi, dijital pazarlamanın iş hayatındaki yeri , anahtar kelimeler : pazarlama, ticaret, dijtalleşme, müşteri, ürün "
input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
outputs = model.generate(input_ids)
print(tokenizer.decode(outputs[0]))