Spaces:
Running
Running
Fix added
Browse files- SkinCancerDiagnosis.py +2 -1
SkinCancerDiagnosis.py
CHANGED
@@ -131,6 +131,7 @@ class SkinDiseaseClassifier:
|
|
131 |
self.base_models = None
|
132 |
self.meta_model = None
|
133 |
self.resnet_feature_extractor = None
|
|
|
134 |
|
135 |
# Image transformations
|
136 |
self.transform = transforms.Compose([
|
@@ -291,7 +292,7 @@ class SkinDiseaseClassifier:
|
|
291 |
with torch.no_grad():
|
292 |
output_multi = self.skincon_model(image_tensor)
|
293 |
probs_multi = torch.sigmoid(output_multi).squeeze().numpy()
|
294 |
-
|
295 |
threshold = 0.5
|
296 |
predicted_labels_multi = [self.multilabel_class_names[i] for i, p in enumerate(probs_multi) if p > threshold]
|
297 |
print("Predicted labels multi : ",predicted_labels_multi)
|
|
|
131 |
self.base_models = None
|
132 |
self.meta_model = None
|
133 |
self.resnet_feature_extractor = None
|
134 |
+
self.skincon_model = None
|
135 |
|
136 |
# Image transformations
|
137 |
self.transform = transforms.Compose([
|
|
|
292 |
with torch.no_grad():
|
293 |
output_multi = self.skincon_model(image_tensor)
|
294 |
probs_multi = torch.sigmoid(output_multi).squeeze().numpy()
|
295 |
+
print(f"Probabilities : {probs_multi}")
|
296 |
threshold = 0.5
|
297 |
predicted_labels_multi = [self.multilabel_class_names[i] for i, p in enumerate(probs_multi) if p > threshold]
|
298 |
print("Predicted labels multi : ",predicted_labels_multi)
|