Update app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 from gradio import FileData
 import time
 import spaces
-ckpt = "meta-llama/Llama-3.2-11B-Vision-Instruct"
+ckpt = "omkarthawakar/LlamaV-o1"
 model = MllamaForConditionalGeneration.from_pretrained(ckpt,
     torch_dtype=torch.bfloat16).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt)