Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,6 @@ class GradioWindow():
|
|
47 |
self.SAM_CHECKPOINT_PATH = SAM_CHECKPOINT_PATH
|
48 |
|
49 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
50 |
-
self.download_weights()
|
51 |
# for debug
|
52 |
# self.augmenter = None
|
53 |
self.augmenter = Augmenter(device=self.device)
|
|
|
47 |
self.SAM_CHECKPOINT_PATH = SAM_CHECKPOINT_PATH
|
48 |
|
49 |
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
50 |
# for debug
|
51 |
# self.augmenter = None
|
52 |
self.augmenter = Augmenter(device=self.device)
|