Andres Johan Florez Gonzalez commited on
Commit
c95ae5c
·
verified ·
1 Parent(s): 464163c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ def main_function(plan_name, apch1, apch6, apch11, coord1, coord6, coord11):
76
  aps_chs = np.zeros((dimension, dimension, len(channels)))
77
 
78
  # Load plan
79
- numero = plan_name[:1]
80
  plan_in = np.array(Image.open(f"{path_main}Scennarios init/Scennarios B/{numero}.png")) / 255
81
 
82
  coords = [coord1, coord6, coord11]
 
76
  aps_chs = np.zeros((dimension, dimension, len(channels)))
77
 
78
  # Load plan
79
+ numero = plan_name.split('.')[0]
80
  plan_in = np.array(Image.open(f"{path_main}Scennarios init/Scennarios B/{numero}.png")) / 255
81
 
82
  coords = [coord1, coord6, coord11]