Update app.py

app.py

@@ -14,7 +14,7 @@ dataset = load_dataset("mwitiderrick/swahili")
 print(f"Dataset columns: {dataset['train'].column_names}")
 
 # Initialize the tokenizer and model
-model_name = "gpt2-small"  # Use a smaller variant of GPT-2 for efficiency
+model_name = "gpt2"  # Use a smaller variant of GPT-2 for efficiency
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)