Update app.py

app.py

@@ -18,8 +18,8 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Load model and processor
-ckpt ="Daemontatox/DocumentCogito"
-model = MllamaForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
+ckpt ="mistral-community/pixtral-12b"
+model = AutoModelForImageTextToText.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
 
 class DocumentState: