add ckpt locations

app.py

@@ -23,6 +23,9 @@ print(f"Is CUDA available: {torch.cuda.is_available()}")
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 # Tesla T4
 
+pretrained_ckpt = "mplugowl7bvideo/"
+trained_ckpt = "owl-con/checkpoint-5178/pytorch_model.bin"
+
 tokenizer = LlamaTokenizer.from_pretrained(pretrained_ckpt)
 image_processor = MplugOwlImageProcessor.from_pretrained(pretrained_ckpt)
 processor = MplugOwlProcessor(image_processor, tokenizer)