Update handler.py

handler.py

@@ -6,8 +6,8 @@ import base64
 
 # Initialize the model and tokenizer
 model_id = "zesquirrelnator/moondream2-finetuneV2"
-model = AutoModelForCausalLM.from_pretrained(model_id)
-tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2")
+model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("vikhyatk/moondream2", trust_remote_code=True)
 
 # Check if CUDA (GPU support) is available and then set the device to GPU or CPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")