Update app.py

app.py

@@ -19,7 +19,7 @@ logger = logging.getLogger(__name__)
 
 # Load model and processor
 ckpt ="Daemontatox/DocumentCogito"
-model = Qwen2VLForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
+model = MllamaForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
 processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
 
 class DocumentState: