Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import pandas as pd
+import numpy as np
 from tensorflow.keras.models import Sequential
 from tensorflow.keras.layers import Dense, Dropout, Activation
 
@@ -41,7 +42,8 @@ ann_model.load_weights('model_weights.h5')
 
 # Make predictions when the 'Predict' button is clicked
 if user_input.form_submit_button('Predict'):
-    prediction = ann_model.predict(input_data)
+    input_array = np.array(input_data)
+    prediction = ann_model.predict(input_array)
     prediction_label = 'Malignant' if prediction[0] >= 0.5 else 'Benign'
     st.subheader('Prediction')
     st.write(f'The lesion is predicted to be: {prediction_label}')