Spaces:
Sleeping
Sleeping
Commit
·
875e0ac
1
Parent(s):
0118b47
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ transform = transforms.Compose([
|
|
33 |
inv_transform = transforms.Normalize(mean=[-2, -2, -2], std=[4, 4, 4])
|
34 |
|
35 |
|
36 |
-
def image_classifier(input_image, top_classes=3, show_cam=True, target_layer=-
|
37 |
input = transform(input_image).unsqueeze(0)
|
38 |
|
39 |
output = model(input)
|
|
|
33 |
inv_transform = transforms.Normalize(mean=[-2, -2, -2], std=[4, 4, 4])
|
34 |
|
35 |
|
36 |
+
def image_classifier(input_image, top_classes=3, show_cam=True, target_layer=-2, transparency=0.7):
|
37 |
input = transform(input_image).unsqueeze(0)
|
38 |
|
39 |
output = model(input)
|