Update model path

handler.py

@@ -8,14 +8,14 @@ class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
         self.model =  AutoModelForCausalLM.from_pretrained(
-            path, device_map="auto", load_in_8bit=True)
+            "tiiuae/falcon-rw-1b", device_map="auto", load_in_8bit=True)
         self.model = PeftModel.from_pretrained(
             self.model,
             "MichaelAI23/falcon-rw-1b_8bit_finetuned",
             torch_dtype=torch.float16,
             device_map="auto"
         )
-        self.tokenizer = AutoTokenizer.from_pretrained(path)
+        self.tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-rw-1b")
         self.template = {
             "prompt_input": "Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n\n### Instruction:\n{instruction}\n\n### Input:\n{input}\n\n### Response:\n",
             "prompt_no_input": "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{instruction}\n\n### Response:\n",