Spaces:
Sleeping
Sleeping
anindya-hf-2002
commited on
Commit
•
a9c06f8
1
Parent(s):
84393df
updated app.py
Browse files
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
|
10 |
-
g_NP_checkpoint = 'models
|
11 |
-
g_PN_checkpoint = 'models
|
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)
|