Spaces:
Runtime error
Runtime error
Update img_extractor.py
Browse files- img_extractor.py +1 -1
img_extractor.py
CHANGED
@@ -27,7 +27,7 @@ class RoiExtractor:
|
|
27 |
def __init__(self):
|
28 |
# Model initialization
|
29 |
self.model = build_image_model()
|
30 |
-
self.model.load_weights(
|
31 |
self.heatmap_model = Model(inputs=self.model.inputs, outputs=self.model.layers[-3].output)
|
32 |
|
33 |
# Outputs
|
|
|
27 |
def __init__(self):
|
28 |
# Model initialization
|
29 |
self.model = build_image_model()
|
30 |
+
self.model.load_weights("https://drive.google.com/file/d/1JqIQHYvW8q5DMjOpcVsfkwxaOZP2CmnL/view?usp=sharing")
|
31 |
self.heatmap_model = Model(inputs=self.model.inputs, outputs=self.model.layers[-3].output)
|
32 |
|
33 |
# Outputs
|