Spaces:
Build error
Build error
Jiayuan Gu
commited on
Commit
·
30c69dc
1
Parent(s):
a123cb5
fix a bug
Browse files
app.py
CHANGED
|
@@ -290,8 +290,13 @@ def set_prompts():
|
|
| 290 |
predictor.patches = patches
|
| 291 |
predictor.input_processor.center = np.array(request_data["center"])
|
| 292 |
predictor.input_processor.scale = request_data["scale"]
|
| 293 |
-
|
| 294 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 295 |
# instance_id = request_data["instance_id"] # int
|
| 296 |
if len(prompt_coords) == 0:
|
| 297 |
predictor.prompt_mask = None
|
|
|
|
| 290 |
predictor.patches = patches
|
| 291 |
predictor.input_processor.center = np.array(request_data["center"])
|
| 292 |
predictor.input_processor.scale = request_data["scale"]
|
| 293 |
+
try:
|
| 294 |
+
if request_data["prompt_mask"] is not None:
|
| 295 |
+
predictor.prompt_mask = torch.tensor(request_data["prompt_mask"]).cuda()
|
| 296 |
+
else:
|
| 297 |
+
predictor.prompt_mask = None
|
| 298 |
+
except:
|
| 299 |
+
predictor.prompt_mask = None
|
| 300 |
# instance_id = request_data["instance_id"] # int
|
| 301 |
if len(prompt_coords) == 0:
|
| 302 |
predictor.prompt_mask = None
|