bug fix
Browse files
app.py
CHANGED
@@ -175,9 +175,9 @@ def predict_and_analyze(model_name, num_channels, dim, image):
|
|
175 |
print(output)
|
176 |
|
177 |
if output[0] < output[1]:
|
178 |
-
output = 'Planet
|
179 |
else:
|
180 |
-
output = 'No planet
|
181 |
|
182 |
return output, input_image, activation_1, activation_2
|
183 |
|
|
|
175 |
print(output)
|
176 |
|
177 |
if output[0] < output[1]:
|
178 |
+
output = 'Planet predicted with %f percent confidence' % (100*output[1])
|
179 |
else:
|
180 |
+
output = 'No planet predicted with %f percent confidence' % (100 - 100*output[0])
|
181 |
|
182 |
return output, input_image, activation_1, activation_2
|
183 |
|