Hugging Face's logo

Spaces:
MedMagik
/
ChexnetKIMS
Running

App Files Community
MedMagik commited on
Commit
49d66c6
·
verified ·
1 Parent(s): 4c4deec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ import numpy as np
8
  from PIL import Image
9
  model = load_model('Densenet.h5')
10
  model.load_weights("pretrained_model.h5")
 
11
  class_names = ['Cardiomegaly', 'Emphysema', 'Effusion', 'Hernia', 'Infiltration', 'Mass', 'Nodule', 'Atelectasis', 'Pneumothorax', 'Pleural_Thickening', 'Pneumonia', 'Fibrosis', 'Edema', 'Consolidation']
12
  def get_gradcam(model, img, layer_name):
13
 
 
8
  from PIL import Image
9
  model = load_model('Densenet.h5')
10
  model.load_weights("pretrained_model.h5")
11
+ layer_name = 'conv5_block16_concat'
12
  class_names = ['Cardiomegaly', 'Emphysema', 'Effusion', 'Hernia', 'Infiltration', 'Mass', 'Nodule', 'Atelectasis', 'Pneumothorax', 'Pleural_Thickening', 'Pneumonia', 'Fibrosis', 'Edema', 'Consolidation']
13
  def get_gradcam(model, img, layer_name):
14