Update app.py

app.py

@@ -4,6 +4,7 @@ from outlines import generate, samplers
 from pydantic import BaseModel, Field
 from typing import Optional
 import json
+import llama_cpp
 
 # Define the output schema
 class Medication(BaseModel):
@@ -18,7 +19,10 @@ class Medication(BaseModel):
     total_daily_dose_mg: Optional[float] = Field(default=None, description="The total daily dose in milligrams.")
 
 # Load your model locally via llama-cpp
-model = llamacpp("cmcmaster/drug_parsing_Llama-3.2-1B-Instruct-Q5_K_S-GGUF", "drug_parsing_llama-3.2-1b-instruct-q5_k_s.gguf")
+model = llamacpp(
+    "cmcmaster/drug_parsing_Llama-3.2-1B-Instruct-Q5_K_S-GGUF", 
+    "drug_parsing_llama-3.2-1b-instruct-q5_k_s.gguf",
+    tokenizer=llama_cpp.llama_tokenizer.LlamaHFTokenizer.from_pretrained("cmcmaster/drug_parsing_Llama-3.2-1B-Instruct")
 
 sampler = samplers.greedy()