book-buddy-question-generator

Book Buddy - Question Generator

This fine-tuned model was generated to help students study. By submitting their texts, the model will generate a question to help them study.

Model Details

• Model Architecture: T5
• Tokenizer Used:
• Language: English
• Task: Question Generation

Model Usage

How to Use

Provide instructions on how to use your model in a clear and concise manner.

# Example code for using the model
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

# Load the tokenizer
tokenizer = AutoTokenizer.from_pretrained("t5_tokenizer")

# Load the model
model = AutoModelForSeq2SeqLM.from_pretrained("t5_trained_model")

# Generate a question
input_text = "Provide a sample input text."
input_ids = tokenizer.encode(input_text, return_tensors="pt", padding=True, max_length=512, truncation=True)

# Generate question
question_ids = model.generate(input_ids, max_length=32, num_return_sequences=1, num_beams=4)

# Decode the generated question
generated_question = tokenizer.decode(question_ids[0], skip_special_tokens=True)
print(f"Generated Question: {generated_question}")