Update handler.py

handler.py

@@ -1,13 +1,13 @@
 import torch
-from transformers import MllamaForConditionalGeneration, AutoProcessor
+from transformers import LlamaForCausalLM, AutoProcessor
 from PIL import Image
 import base64
 import io
 
 # Load model and processor globally
-model_id = "meta-llama/Llama-3.2-90B-Vision-Instruct"
+model_id = "kiddobellamy/Llama_Vision"
 
-model = MllamaForConditionalGeneration.from_pretrained(
+model = LlamaForCausalLM.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
@@ -49,4 +49,3 @@ def handler(event, context):
 
     except Exception as e:
         return {'error': str(e)}
-#111