Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import multiprocessing
 
-model_id = "EleutherAI/gpt-neo-125M"
+model_id = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id)
 generator = pipeline("text-generation", model=model, tokenizer=tokenizer)