JarvisLabs commited on
Commit
ec07d4b
·
verified ·
1 Parent(s): 74ef4ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -64,11 +64,11 @@ def create_zip(files,captions,trigger):
64
  captions=captions.split("\n")
65
  #cute files and "tags:"
66
  captions= [cap.split("file:")[0][5:] for cap in captions]
67
- temp_dir="/content"
68
- os.makedirs(temp_dir, exist_ok=True)
69
  # Create a zip file
70
  #os.makedirs(temp_dir, exist_ok=True)
71
- zip_path = os.path.join(temp_dir, "training_data.zip")
72
  with zipfile.ZipFile(zip_path, "w") as zip_file:
73
  for i, file in enumerate(files):
74
  # Add image to zip
 
64
  captions=captions.split("\n")
65
  #cute files and "tags:"
66
  captions= [cap.split("file:")[0][5:] for cap in captions]
67
+ #temp_dir="/content"
68
+ #os.makedirs(temp_dir, exist_ok=True)
69
  # Create a zip file
70
  #os.makedirs(temp_dir, exist_ok=True)
71
+ zip_path = "training_data.zip" #os.path.join(temp_dir, "training_data.zip")
72
  with zipfile.ZipFile(zip_path, "w") as zip_file:
73
  for i, file in enumerate(files):
74
  # Add image to zip