Update main.py

main.py

@@ -3,7 +3,8 @@ from fastapi import FastAPI, Form
 from pydantic import BaseModel
 
 #Model loading
-llm = AutoModelForCausalLM.from_pretrained(model_file="Meta-Llama-3-8B-Instruct-Q4_K_M.gguf",
+llm = AutoModelForCausalLM.from_pretrained("TheBloke/CodeLlama-7B-Python-GGUF",
+model_file="Meta-Llama-3-8B-Instruct-Q4_K_M.gguf",
 model_type='llama',
 # max_new_tokens = 1096,
 threads = 3,