Update main.py

main.py

@@ -17,7 +17,7 @@ os.makedirs(cache_dir, exist_ok=True)
 os.chmod(cache_dir, 0o777)
 
 # Load model and tokenizer
-model_name = "microsoft/DialoGPT-small"
+model_name = "EleutherAI/gpt-neo-1.3B"
 tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
 model = AutoModelForCausalLM.from_pretrained(model_name, cache_dir=cache_dir)