Spaces:
Build error
Build error
bug fix
Browse files
app.py
CHANGED
|
@@ -175,7 +175,10 @@ def predict_and_analyze(model_name, num_channels, dim, image):
|
|
| 175 |
print("Looking at activations")
|
| 176 |
output, activation_1, activation_2 = get_activations(model, image, sub_mean=True)
|
| 177 |
|
| 178 |
-
output
|
|
|
|
|
|
|
|
|
|
| 179 |
|
| 180 |
return output, input_image, activation_1, activation_2
|
| 181 |
|
|
|
|
| 175 |
print("Looking at activations")
|
| 176 |
output, activation_1, activation_2 = get_activations(model, image, sub_mean=True)
|
| 177 |
|
| 178 |
+
if output[0] < output[1]:
|
| 179 |
+
output = 'Planet prediction with %f percent confidence' % (100*output[1])
|
| 180 |
+
else:
|
| 181 |
+
output = 'No planet prediction with %f percent confidence' % (100 - 100*output[0])
|
| 182 |
|
| 183 |
return output, input_image, activation_1, activation_2
|
| 184 |
|