Update app.py

app.py

@@ -107,8 +107,9 @@ def download_button(object_to_download, download_filename, button_text, file_ext
 file_upload = st.file_uploader("Upload a csv file", type="csv")
 if file_upload is not None:
     data = pd.read_csv(file_upload)
-    column = data["S11"].iloc[1:].values
-    with open("automl3.pkl", "rb") as f:
+    #column = data["S11"].iloc[1:].values
+    column = data["S11"].values.reshape(1,-1)
+    with open("automl4.pkl", "rb") as f:
         model = pickle.load(f)
     pred_clip = model.predict([column])
     pred_clip = np.clip(pred_clip, [0.2,0.4,3.9,0.2,13.9,13.8,13.2],[1.01,1.21,4.71,0.8,14.701,14.201,14.001])