Update app.py

app.py

@@ -71,7 +71,7 @@ with gr.Blocks() as demo:
         input_array = np.expand_dims(resized_array, axis=0) # add an extra dimension to represent the batch size
 
         # Make a prediction using the model
-        prediction = probability_model.predict(img_array)
+        prediction = probability_model.predict(input_array)
 
         # Postprocess the prediction and return it
         return np.argmax(predictions[0])