Update text2img.py

Author: Beyondo

text2img.py

@@ -1,4 +1,4 @@
-import torch, os, datetime, colab
+import torch, os, time, datetime, colab
 from IPython.display import Image
 from IPython.display import display
 def process(seed, positive_prompt, negative_prompt, guidance_scale, inference_steps, save_to_google_drive, directory):
@@ -8,8 +8,7 @@ def process(seed, positive_prompt, negative_prompt, guidance_scale, inference_st
     if save_to_google_drive:
         dir = '/content/gdrive/MyDrive/' + directory
         if not os.path.exists(dir): os.makedirs(dir)
-        num = len([name for name in os.listdir(dir) if os.path.isfile(os.path.join(dir, name))])
-        imgSavePath = "%s/%d-voidops" % (dir, num)
+        imgSavePath = "%s/%d-voidops" % (dir, int(time.mktime(datetime.now().timetuple())))
         image.save(imgSavePath + ".png")
         print("Saved to " + imgSavePath)
         display(image)