trial

app.py

@@ -32,13 +32,15 @@ def prediction():
       df = pd.read_csv(uploaded_file)
       model = load_model()
       prediction = predict_model(model, df).drop(columns='Unnamed: 0')
+      return prediction
+   
+result = prediction()
 
-      # show the result
-      st.write(prediction)
+# show the result
+st.write(result)
 
-      # download the result
-      csv = convert_df(prediction)
-      if st.download_button('Download Prediction', csv, 'prediction.csv'):
-         st.write('Thanks for downloading!')
+# download the result
+csv = convert_df(result)
+if st.download_button('Download Prediction', csv, 'prediction.csv'):
+   st.write('Thanks for downloading!')
 
-prediction()