OmerFarooq commited on
Commit
9b0db10
1 Parent(s): 2aeb247

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def gen_grad_both(grad_model, img):
136
 
137
  return img_c, img_d, y_pred_c, infer
138
 
139
- weights = "weights.h5"
140
 
141
  def get_grad(img):
142
  img = img_pros(img)
 
136
 
137
  return img_c, img_d, y_pred_c, infer
138
 
139
+ weights = "weights_nm.h5"
140
 
141
  def get_grad(img):
142
  img = img_pros(img)