Update app.py

app.py

@@ -128,9 +128,9 @@ class DatasetBuilder:
                 dataset_file_path = os.path.join(DATA_DIR, dataset_file)
                 with open(dataset_file_path, 'w', newline='') as csvfile:
                     writer = csv.writer(csvfile)
-                    writer.writerow(['image', 'text'])
+                    writer.writerow(['image', 'tags'])
                     for item in self.dataset:
-                        writer.writerow([item['image'], item['text']])
+                        writer.writerow([item['image'], item['tags']])
                 zipf.write(dataset_file_path, os.path.basename(dataset_file_path))
                 
                 # Add all images