acmyu commited on
Commit
4891e68
·
1 Parent(s): fc6eedb

fix upload filename

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -83,9 +83,11 @@ max_batch_size = 8
83
 
84
 
85
  def save_temp_imgs(imgs):
 
 
86
  for img in imgs:
87
 
88
- img_name = str(uuid.uuid4())+'.png'
89
  img.save(img_name)
90
  print(img_name)
91
 
 
83
 
84
 
85
  def save_temp_imgs(imgs):
86
+ os.makedirs('temp', exist_ok=True)
87
+
88
  for img in imgs:
89
 
90
+ img_name = '@temp/'+str(uuid.uuid4())+'.png'
91
  img.save(img_name)
92
  print(img_name)
93