bug fix

app.py

@@ -313,10 +313,10 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
                                                                       sub_mean=True)
     print("Activations and predictions finished")
     
-    if output[0] < output[1]:
-        output = 'Planet predicted with %.3f percent confidence' % (100*output[1])
+    if output[0][0] < output[0][1]:
+        output = 'Planet predicted with %.3f percent confidence' % (100*output[0][1])
     else:
-        output = 'No planet predicted with %.3f percent confidence' % (100*output[0])
+        output = 'No planet predicted with %.3f percent confidence' % (100*output[0][0])
         
     input_image = normalize_array(input_image)
     activation_1 = normalize_array(activation_1)