Update main.py

main.py

@@ -4,10 +4,10 @@ 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")
-app = fastapi.FastAPI(title="WizardCoder")
+llm = AutoModelForCausalLM.from_pretrained("TheBloke/cassandra-6.9B-GGML",
+                                           model_file="cassandra-6.9b.ggmlv0.q4_0.bin",
+                                           model_type="gpt_neox")
+app = fastapi.FastAPI(title="cassandra-6.9b")
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],