Update prediction.py

prediction.py

@@ -60,27 +60,28 @@ def run():
     data_inf = pd.DataFrame([data_inf])
     st.dataframe(data_inf)
 
-    # Split between Numerical Columns and Categorical Columns
-
-    data_inf_num = data_inf[list_num_cols]
-    data_inf_cat = data_inf[list_cat_cols]
-
-    # Feature Scaling and Feature Encoding
-
-    data_inf_num_scaled = model_scaler.transform(data_inf_num)
-    data_inf_final = np.concatenate([data_inf_num_scaled], axis=1)
-
-    # Prediksi menggunakan random forest
-
-    y_pred_inf = loaded_model.predict(data_inf_final)
-
-    # Mengubah hasil prediksi menjadi label yang sesuai
-    if y_pred_inf == 0:
-        hasil_prediksi = "Ada Harapan"
-    else:
-        hasil_prediksi = "Death"
-
-    st.write('# Kemungkinan : ', hasil_prediksi)
+    if submitted:
+        # Split between Numerical Columns and Categorical Columns
+    
+        data_inf_num = data_inf[list_num_cols]
+        data_inf_cat = data_inf[list_cat_cols]
+    
+        # Feature Scaling and Feature Encoding
+    
+        data_inf_num_scaled = model_scaler.transform(data_inf_num)
+        data_inf_final = np.concatenate([data_inf_num_scaled], axis=1)
+    
+        # Prediksi menggunakan random forest
+    
+        y_pred_inf = loaded_model.predict(data_inf_final)
+    
+        # Mengubah hasil prediksi menjadi label yang sesuai
+        if y_pred_inf == 0:
+            hasil_prediksi = "Ada Harapan"
+        else:
+            hasil_prediksi = "Death"
+    
+        st.write('# Kemungkinan : ', hasil_prediksi)
 
 if __name__== '__main__':
     run()