zhiweili commited on
Commit
a425cf8
·
1 Parent(s): 4279b74

fix output dir not found

Browse files
Files changed (1) hide show
  1. segment_utils.py +7 -1
segment_utils.py CHANGED
@@ -29,7 +29,13 @@ def restore_result(croper, category, generated_image):
29
  # else:
30
  # extension = 'jpg'
31
 
32
- path = f"output/{uuid.uuid4()}.{extension}"
 
 
 
 
 
 
33
  restored_image.save(path, quality=100)
34
 
35
  return restored_image, path
 
29
  # else:
30
  # extension = 'jpg'
31
 
32
+ tmpPrefix = "/tmp/gradio/"
33
+
34
+ targetDir = f"{tmpPrefix}output/"
35
+ if not os.path.exists(targetDir):
36
+ os.makedirs(targetDir)
37
+
38
+ path = f"{targetDir}{uuid.uuid4()}.{extension}"
39
  restored_image.save(path, quality=100)
40
 
41
  return restored_image, path