MBA98 commited on
Commit
9beee6f
1 Parent(s): 78450f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def load_vgg16_model():
109
  'F1Score': tfa.metrics.F1Score,
110
  'MultiLabelConfusionMatrix': tfa.metrics.MultiLabelConfusionMatrix
111
  })
112
- return vgg16_bs32_img128_CLAHE4
113
 
114
 
115
  def predict_input_image(img):
 
109
  'F1Score': tfa.metrics.F1Score,
110
  'MultiLabelConfusionMatrix': tfa.metrics.MultiLabelConfusionMatrix
111
  })
112
+ return vgg16_DR
113
 
114
 
115
  def predict_input_image(img):