vcabreragreco commited on
Commit
6f15783
·
verified ·
1 Parent(s): 9a56abb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,8 +23,8 @@ dx_type_map={'histo':1, 'follow_up':2, 'consensus':3, 'confocal':4}
23
  def classify_img(image,age=60, sex='male', localization='back', dx_type='follow-up'):
24
  img_processed = preprocess_image(image)
25
  age_clean = int(age)
26
- sex_clean=sex_map.get(sex.lower())
27
- dx_type=dx_type_map.get(dx_type_map.lower())
28
  localization_clean=localization_map.get(localization.lower(),0)
29
  meta_input= np.array([[age_clean,sex_clean,localization_clean,dx_type]])
30
 
 
23
  def classify_img(image,age=60, sex='male', localization='back', dx_type='follow-up'):
24
  img_processed = preprocess_image(image)
25
  age_clean = int(age)
26
+ sex_clean=sex_map.get(sex.lower(),1)
27
+ dx_type=dx_type_map.get(dx_type_map.lower(),0)
28
  localization_clean=localization_map.get(localization.lower(),0)
29
  meta_input= np.array([[age_clean,sex_clean,localization_clean,dx_type]])
30