Update app.py

app.py

@@ -6,7 +6,7 @@ model = tf.keras.models.load_model("./model.h5")
 
 #return a dictionary of labels and probabilities
 def fire_or_non_fire(img):
-    img = img.reshape((-1 ,105, 165, 3)))
+    img = img.reshape((-1 ,105, 165, 3))
     prediction = model.predict(img).tolist()[0]
     labels = ["Fire", "Non Fire"]
     return {labels[i]: prediction[i] for i in range(2)}