Update app.py

app.py

@@ -107,15 +107,15 @@ 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
-    #column = data["S11"].values
+    #column = data["S11"].iloc[1:].values
+    column = data["S11"].values
     column = column.reshape(1,-1)
     st.write(column.shape)
-    with open("automl3.pkl", "rb") as f:
+    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])
-    predictions = pd.DataFrame(pred_clip.tolist(), columns = ["w1","w2","w3","s1","l1","l2","l3"])
+    predictions = pd.DataFrame(pred_clip[0].tolist(), columns = ["w1","w2","w3","s1","l1","l2","l3"])
     
     is_download = st.checkbox("Download predictions", value=False)
     if is_download: