Update app.py

app.py

@@ -749,8 +749,8 @@ def submit_predict(predict_filepath, task, preset, target_family, opts, state):
     annotated_df = orig_df[~orig_df['Y'].isna()].copy()
     annotated_df.rename(columns={'Y': 'Y^'}, inplace=True)
     annotated_df['Source'] = 'Database'
-    # Resave the unannotated data
-    unannotated_df = orig_df[orig_df['Y'].isna()].drop(['Y', 'Target Family'], axis=1)
+    # Save the unannotated data
+    unannotated_df = orig_df[orig_df['Y'].isna()].drop(['Y'], axis=1)
     if not unannotated_df.empty:
         unannotated_df.to_csv(predict_filepath, index=False, na_rep='')
     else: