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
|