Jiayuan Gu commited on
Commit
30c69dc
1 Parent(s): a123cb5
Files changed (1) hide show
  1. app.py +7 -2
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
- if request_data["prompt_mask"] is not None:
294
- predictor.prompt_mask = torch.tensor(request_data["prompt_mask"]).cuda()
 
 
 
 
 
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