anindya-hf-2002 commited on
Commit
a9c06f8
1 Parent(s): 84393df

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,9 +6,9 @@ from src.models import ResUNetGenerator
6
  from src.explainer import GradCAM, preprocess_image
7
 
8
  # Loading Models
9
- classifier_path = 'models\efficientnet_b1-epoch16-val_loss0.46_ft.ckpt'
10
- g_NP_checkpoint = 'models\g_NP_best.ckpt'
11
- g_PN_checkpoint = 'models\g_PN_best.ckpt'
12
  g_NP = load_model(g_NP_checkpoint, ResUNetGenerator(gf=32, channels=1))
13
  g_PN = load_model(g_PN_checkpoint, ResUNetGenerator(gf=32, channels=1))
14
  classifier = load_classifier(classifier_path)
 
6
  from src.explainer import GradCAM, preprocess_image
7
 
8
  # Loading Models
9
+ classifier_path = './models/efficientnet_b1-epoch16-val_loss0.46_ft.ckpt'
10
+ g_NP_checkpoint = './models/g_NP_best.ckpt'
11
+ g_PN_checkpoint = './models/g_PN_best.ckpt'
12
  g_NP = load_model(g_NP_checkpoint, ResUNetGenerator(gf=32, channels=1))
13
  g_PN = load_model(g_PN_checkpoint, ResUNetGenerator(gf=32, channels=1))
14
  classifier = load_classifier(classifier_path)