MikkoLipsanen commited on
Commit
a4c9237
·
verified ·
1 Parent(s): 7926f67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def get_segmenter():
27
  """Initialize segmentation class."""
28
  try:
29
  segmenter = SegmentImage(line_model_path=LINE_MODEL_PATH,
30
- device='0',
31
  line_iou=0.3,
32
  region_iou=0.5,
33
  line_overlap=0.5,
 
27
  """Initialize segmentation class."""
28
  try:
29
  segmenter = SegmentImage(line_model_path=LINE_MODEL_PATH,
30
+ device='cuda:0',
31
  line_iou=0.3,
32
  region_iou=0.5,
33
  line_overlap=0.5,