Spaces:

miojizzy
/

video_tools

Sleeping

miojizzy commited on Oct 15, 2023

Commit

269a7d4

1 Parent(s): b42cdae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def make_dataset(vc, df, input_w1, input_w2, input_h1, input_h2):
     path = os.path.dirname(__file__)+"/"+str(uuid.uuid1())
     os.mkdir(path)
     os.mkdir(path+"/train")
-    headers = ["image_name", "label"]
     label_df = pd.DataFrame(columns=headers)
     with ZipFile(path+"/train.zip", "w") as zf:
         for line in df.itertuples():

     path = os.path.dirname(__file__)+"/"+str(uuid.uuid1())
     os.mkdir(path)
     os.mkdir(path+"/train")
+    headers = ["name", "label"]
     label_df = pd.DataFrame(columns=headers)
     with ZipFile(path+"/train.zip", "w") as zf:
         for line in df.itertuples():