Update app.py

app.py

@@ -1057,9 +1057,7 @@ def submit_predict(predict_filepath, task, preset, target_family, opts, job_info
 
         prediction_df = apply_advanced_opts(prediction_df, opts, df_training)
 
-        prediction_df.drop(
-            [col for col in ['N', 'FP'] if col in prediction_df.columns], axis=1
-        ).to_csv(predictions_file, index=False, na_rep='')
+        prediction_df.drop(['N', 'FP'], axis=1, errors='ignore').to_csv(predictions_file, index=False, na_rep='')
         status = 'COMPLETED'
 
         return {run_state: False}