Update app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 
 
 
-model_name = "gpt2-small"
+model_name = "gpt2"
 model = GPT2LMHeadModel.from_pretrained(model_name)
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 
@@ -33,12 +33,12 @@ data_collator = DataCollatorForLanguageModeling(
 training_args = TrainingArguments(
     output_dir=output_dir,
     overwrite_output_dir=True,
-    num_train_epochs=10,
-    per_device_train_batch_size=2,
+    num_train_epochs=20,
+    per_device_train_batch_size=1,
     save_steps=10_000,
     save_total_limit=2,
     logging_dir=output_dir,
-    logging_steps=100,
+    logging_steps=500,
     report_to=[],
 )