kumahiyo commited on
Commit
ea488ba
1 Parent(s): d08fd88

add image save path

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -44,7 +44,7 @@ def draw(data: Data):
44
  image = pipe(prompt, negative_prompt=n_prompt).images[0]
45
 
46
  fileName = "sd_" + str(time.time()) + '.png'
47
- # image.save("/code/tmpdir/" + fileName)
48
 
49
  print(fileName)
50
 
 
44
  image = pipe(prompt, negative_prompt=n_prompt).images[0]
45
 
46
  fileName = "sd_" + str(time.time()) + '.png'
47
+ image.save("/code/tmpdir/" + fileName)
48
 
49
  print(fileName)
50