Update main.py

main.py

@@ -4,9 +4,9 @@ from fastapi.middleware.cors import CORSMiddleware
 from ctransformers import AutoModelForCausalLM
 from pydantic import BaseModel
 
-llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardCoder-15B-1.0-GGML",
-                                           model_file="WizardCoder-15B-1.0.ggmlv3.q4_0.bin",
-                                           model_type="starcoder")
+llm = AutoModelForCausalLM.from_pretrained("TheBloke/wizardLM-7B-GGML",
+                                           model_file="wizardLM-7B.ggmlv3.q4_K_M.bin",
+                                           model_type="llama")
 app = fastapi.FastAPI(title="WizardCoder")
 app.add_middleware(
     CORSMiddleware,