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/cassandra-6.9B-GGML",
-                                           model_file="cassandra-6.9b.ggmlv0.q4_0.bin",
-                                           model_type="gpt_neox")
-app = fastapi.FastAPI(title="cassandra-6.9b")
+llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardLM-7B-V1.0-Uncensored-GGML",
+                                           model_file="wizardlm-7b-v1.0-uncensored.ggmlv3.q2_K.bin",
+                                           model_type="llama")
+app = fastapi.FastAPI(title="WizardLM-7B-V1.0-Uncensored")
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],