Update handler.py

handler.py

@@ -18,7 +18,7 @@ class EndpointHandler:
         # If a model_dir is provided by the endpoint, it will load from that path;
         # otherwise, it uses a default UnsLoth Mistral model.
         self.model, self.tokenizer = FastLanguageModel.from_pretrained(
-            model_name=model_dir if model_dir else "unsloth/mistral-7b-instruct-v0.3-bnb-4bit",
+            model_name=RichardLu/Mistral7b_AE_res,
             max_seq_length=max_seq_length,
             dtype=dtype,
             load_in_4bit=load_in_4bit,