Update generate_csv.py

utils/generate_csv.py

@@ -1,4 +1,3 @@
-from msilib.schema import Directory
 import os 
 import pandas as pd 
 from sklearn.utils import shuffle
@@ -34,15 +33,12 @@ def generate_csv(path):
 
             for filename in os.listdir(directory):
 
-                path = directory + "/" + filename  #required path 
-                data.append([ filename , path  , unique])
+                paths = directory + "/" + filename  #required path 
+                data.append([ filename , paths  , unique])
 
     df = pd.DataFrame(data, columns = ["filename" ,"path", "class"]) 
     df = shuffle(df)
     name = "csv_files/" + "Data-full"        #required path 
     df.to_csv(name, index = False)
     print("Generation Complete")
-    return df    
-
-
-
+    return df