Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|