AAAAAAyq commited on
Commit
c2bd1fd
·
1 Parent(s): 1472255

Add more cached examples

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -157,7 +157,7 @@ def fast_show_mask_gpu(annotation, ax,
157
 
158
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
159
 
160
- def predict(input, input_size=512, high_visual_quality=False):
161
  input_size = int(input_size) # 确保 imgsz 是整数
162
  results = model(input, device=device, retina_masks=True, iou=0.7, conf=0.25, imgsz=input_size)
163
  fig = fast_process(annotations=results[0].masks.data,
 
157
 
158
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
159
 
160
+ def predict(input, input_size=512, high_visual_quality=True):
161
  input_size = int(input_size) # 确保 imgsz 是整数
162
  results = model(input, device=device, retina_masks=True, iou=0.7, conf=0.25, imgsz=input_size)
163
  fig = fast_process(annotations=results[0].masks.data,