JoshMe1 commited on
Commit
31a5c13
·
1 Parent(s): 0073e41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -76,6 +76,13 @@ def main():
76
  if st.button("Predict"):
77
  predictions_df = predict_hotel_price(train_features_path, train_label_path, test_features_path)
78
  st.write(predictions_df)
 
 
 
 
 
 
79
 
80
  if __name__ == '__main__':
81
  main()
 
 
76
  if st.button("Predict"):
77
  predictions_df = predict_hotel_price(train_features_path, train_label_path, test_features_path)
78
  st.write(predictions_df)
79
+ st.download_button(
80
+ label="Download Predictions",
81
+ data=predictions_df.to_csv(index=False),
82
+ file_name="predictions.csv",
83
+ mime="text/csv"
84
+ )
85
 
86
  if __name__ == '__main__':
87
  main()
88
+