Updated checkpoint path and added arXiv badge to README

app.py

@@ -14,11 +14,13 @@ from gpt4v import prompt_image
 HOME = os.getenv("HOME")
 DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 
-SAM_CHECKPOINT = os.path.join(HOME, "app/weights/sam_vit_h_4b8939.pth")
-# SAM_CHECKPOINT = "weights/sam_vit_h_4b8939.pth"
+# SAM_CHECKPOINT = os.path.join(HOME, "app/weights/sam_vit_h_4b8939.pth")
+SAM_CHECKPOINT = "weights/sam_vit_h_4b8939.pth"
 SAM_MODEL_TYPE = "vit_h"
 
 MARKDOWN = """
+[![arXiv](https://img.shields.io/badge/arXiv-1703.06870v3-b31b1b.svg)](https://arxiv.org/pdf/2310.11441.pdf)
+
 <h1 style='text-align: center'>
     <img 
         src='https://som-gpt4v.github.io/website/img/som_logo.png'