Nigerian Food GPT-2 Model
This model is fine-tuned on a dataset of Nigerian food items. It can generate descriptions of Nigerian dishes, including ingredients, health information, and regional origin.
Model Details
- Fine-tuned from: GPT-2 (small)
- Type: Language Model
- Language: English
- License: MIT
Intended Use
This model is intended for generating descriptions of Nigerian food items. It can be used for educational purposes, menu generation, or exploring Nigerian cuisine.
Training Data
The model was fine-tuned on a dataset of 100+ Nigerian food items, including information on the food's name, main ingredients, description, health rating, food class, region of origin, spice level, and price range.
Limitations
- The model's knowledge is limited to the training data provided.
- It may occasionally generate inaccurate or inconsistent information.
- The model should not be used as a definitive source for nutritional or health information.
Usage
'''python from transformers import GPT2LMHeadModel, GPT2Tokenizer
model = GPT2LMHeadModel.from_pretrained("your-username/nigerian-food-gpt2") tokenizer = GPT2Tokenizer.from_pretrained("your-username/nigerian-food-gpt2")
prompt = "Name: Jollof Rice" input_ids = tokenizer.encode(prompt, return_tensors="pt") output = model.generate(input_ids, max_length=200, num_return_sequences=1, no_repeat_ngram_size=2) generated_text = tokenizer.decode(output[0], skip_special_tokens=True) print(generated_text)