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}")
