Spaces:
Build error
Build error
fix SAM_LOCAL bug
Browse files
app.py
CHANGED
|
@@ -151,6 +151,8 @@ def launch(port):
|
|
| 151 |
if SAM_LOCAL:
|
| 152 |
sam_predictor = sam_init(SAM_CKPT_PATH, gpu)
|
| 153 |
print("load sam ckpt done.")
|
|
|
|
|
|
|
| 154 |
|
| 155 |
with gr.Blocks(
|
| 156 |
title="TGS - Demo"
|
|
|
|
| 151 |
if SAM_LOCAL:
|
| 152 |
sam_predictor = sam_init(SAM_CKPT_PATH, gpu)
|
| 153 |
print("load sam ckpt done.")
|
| 154 |
+
else:
|
| 155 |
+
sam_predictor = None
|
| 156 |
|
| 157 |
with gr.Blocks(
|
| 158 |
title="TGS - Demo"
|