Spaces:

miojizzy
/

video_tools

Sleeping

miojizzy commited on Oct 9, 2023

Commit

460bbce

1 Parent(s): 6f64a48

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,12 +46,11 @@ def make_dataset(vc, df):
                 zf.write(path+"/train/"+name, arcname="train/"+name)
                 os.remove(path+"/train/"+name)
                 label_df = pd.concat([label_df, pd.DataFrame([[name, label]], columns=headers)])
-    label_df.to_csv(path+'/label.csv')
     return [path+"/label.csv", path+"/train.zip"]
 with gr.Blocks() as demo:
     with gr.Tab("Create Image Classify Dataset"):
         with gr.Row():

                 zf.write(path+"/train/"+name, arcname="train/"+name)
                 os.remove(path+"/train/"+name)
                 label_df = pd.concat([label_df, pd.DataFrame([[name, label]], columns=headers)])
+    label_df.to_csv(path+'/label.csv', index=False)
     return [path+"/label.csv", path+"/train.zip"]
 with gr.Blocks() as demo:
     with gr.Tab("Create Image Classify Dataset"):
         with gr.Row():