Hugging Face's logo

Spaces:
SebastianBravo
/
SIMCI
Paused

App Files Community
SebastianBravo commited on
Commit
7d8644d
·
1 Parent(s): ded489b

Added new features

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ with tf.device("cpu:0"):
44
  model_unet = utils.import_3d_unet(path_3d_unet)
45
 
46
  # MedNet
47
- device_ids = [7]
48
  mednet_model = utils.create_mednet(weight_path, device_ids)
49
 
50
  # SVM model
 
44
  model_unet = utils.import_3d_unet(path_3d_unet)
45
 
46
  # MedNet
47
+ device_ids = [0]
48
  mednet_model = utils.create_mednet(weight_path, device_ids)
49
 
50
  # SVM model