bug fix
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ def get_activations(intermediate_model, image: list,
|
|
| 68 |
output_1 = outputs[1]
|
| 69 |
output_2 = outputs[2]
|
| 70 |
|
| 71 |
-
output = outputs[0]
|
| 72 |
output = special.softmax(output)
|
| 73 |
|
| 74 |
# origin = 'lower'
|
|
|
|
| 68 |
output_1 = outputs[1]
|
| 69 |
output_2 = outputs[2]
|
| 70 |
|
| 71 |
+
output = outputs[0][0]
|
| 72 |
output = special.softmax(output)
|
| 73 |
|
| 74 |
# origin = 'lower'
|