Updated checkpoint path and added arXiv badge to README
Browse files
app.py
CHANGED
|
@@ -14,8 +14,7 @@ from gpt4v import prompt_image
|
|
| 14 |
HOME = os.getenv("HOME")
|
| 15 |
DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 16 |
|
| 17 |
-
|
| 18 |
-
SAM_CHECKPOINT = "weights/sam_vit_h_4b8939.pth"
|
| 19 |
SAM_MODEL_TYPE = "vit_h"
|
| 20 |
|
| 21 |
MARKDOWN = """
|
|
|
|
| 14 |
HOME = os.getenv("HOME")
|
| 15 |
DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 16 |
|
| 17 |
+
SAM_CHECKPOINT = os.path.join(HOME, "app/weights/sam_vit_h_4b8939.pth")
|
|
|
|
| 18 |
SAM_MODEL_TYPE = "vit_h"
|
| 19 |
|
| 20 |
MARKDOWN = """
|