AbrarFahim75/t5-multitask-book

T5 Multitask Model for Book Genre, Rating, and Title Tasks

This model was trained on a custom dataset of book descriptions and titles. It supports:

• genre: — classify the genre of a book
• rating: — predict the numeric rating
• title: — generate a book title

Usage

from transformers import T5Tokenizer, T5ForConditionalGeneration

model = T5ForConditionalGeneration.from_pretrained("AbrarFahim75/t5-multitask-book")
tokenizer = T5Tokenizer.from_pretrained("AbrarFahim75/t5-multitask-book")

input_text = "genre: A dark and stormy night in an abandoned castle."
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))