Update server.py

server.py

@@ -4,9 +4,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 app = FastAPI()  
 
 # Load Fine-Tuned Model  
-model_name = "/app/models/fine-tuned"  
-tokenizer = AutoTokenizer.from_pretrained("NousResearch/Nous-Hermes-2-Mistral-7B")
-model = AutoModelForCausalLM.from_pretrained("NousResearch/Nous-Hermes-2-Mistral-7B")  
+model_name = "PygmalionAI/pygmalion-7b"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)  
 
 @app.get("/chat")  
 def chat(msg: str):