Huiwenshi commited on
Commit
d160556
·
verified ·
1 Parent(s): a7e6d77

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. hg_app.py +3 -1
hg_app.py CHANGED
@@ -120,7 +120,9 @@ def _gen_shape(
120
  req: gr.Request,
121
  ):
122
  if caption: print('prompt is', caption)
123
- save_folder = os.path.join(SAVE_DIR, str(req.session_hash))
 
 
124
  stats = {}
125
  time_meta = {}
126
  start_time_0 = time.time()
 
120
  req: gr.Request,
121
  ):
122
  if caption: print('prompt is', caption)
123
+ save_folder = os.path.join(SAVE_DIR, str(req.session_hash))
124
+ os.makedirs(save_folder, exist_ok=True)
125
+
126
  stats = {}
127
  time_meta = {}
128
  start_time_0 = time.time()