Update main.py

main.py

@@ -18,7 +18,7 @@ os.makedirs(cache_dir, exist_ok=True)
 os.chmod(cache_dir, 0o777)
 
 # Load model and tokenizer
-model_name = "EleutherAI/gpt-neo-125m"
+model_name = "Qwen/Qwen2.5-0.5B-Instruct"
 tokenizer = AutoTokenizer.from_pretrained(
     model_name,
     trust_remote_code=True,