Update app.py

app.py

@@ -130,13 +130,14 @@ if file_upload is not None:
     #predictions = pd.DataFrame(pred_clip.tolist(), columns = ["w1","w2","w3","s1","l1","l2","l3"])
 
     r2_score = r2_score(best_column, predictions.squeeze())
+    st.write(r2_score)
     #st.write(r2_score)
     if r2_score < 0.98:
         predictions = perturb_array(best_column)
         st.write(r2_score)
         st.write(predictions)
         st.write(type(predictions))
-        predictions = pd.DataFrame([predictions], columns = ["w1","w2","w3","s1","l1","l2","l3"])
+        predictions = pd.DataFrame([predictions.tolist()], columns = ["w1","w2","w3","s1","l1","l2","l3"])
     
     is_download = st.checkbox("Download predictions", value=False)
     if is_download: