Update server.py

server.py

@@ -5,8 +5,8 @@ app = FastAPI()
 
 # Load Fine-Tuned Model  
 model_name = "/app/models/fine-tuned"  
-tokenizer = AutoTokenizer.from_pretrained("TheBloke/Pygmalion-7B-GPTQ")  
-model = AutoModelForCausalLM.from_pretrained(model_name)  
+tokenizer = AutoTokenizer.from_pretrained("NousResearch/Nous-Hermes-2-Mistral-7B")
+model = AutoModelForCausalLM.from_pretrained("NousResearch/Nous-Hermes-2-Mistral-7B")  
 
 @app.get("/chat")  
 def chat(msg: str):