Update app.py

app.py

@@ -74,7 +74,7 @@ with gr.Blocks() as demo:
         prediction = probability_model.predict(input_array)
 
         # Postprocess the prediction and return it
-        return np.argmax(predictions)
+        return np.argmax(prediction)
         
 
     # Creates the Gradio interface objects