add seed no

main.py

@@ -30,13 +30,18 @@ def draw(data: Data):
 	if data.member_secret != "" and data.member_secret == os.environ.get("MEMBER_SECRET"):
 		print(f"Is CUDA available: {torch.cuda.is_available()}")
 
+		seedno = 0
 		if '_seed' in data.string:
 			seed = 1024
+			stext = re.search(r'_seed[1-4]?', text)
+			stext = re.sub('_seed', '', stext.group())
+			if stext.isnumeric() and 0 < int(stext) < 5:
+				seedno = stext
 		else:
 		 	seed = random.randrange(1024)
 
 		text = re.sub('^#', '', data.string)
-		text = re.sub('_seed', '', text)
+		text = re.sub('_seed[1-4]?', '', text)
 
 		# prompt = '(('+text+')) (( photograph )), highly detailed, sharp focus, 8k, 4k, (( photorealism )), detailed, saturated, portrait, 50mm, F/2.8, 1m away, ( global illumination, studio light, volumetric light ), ((( multicolor lights )))'
 		prompt = '(('+text+')) (( photograph )), highly detailed, sharp focus, 8k, 4k, (( photorealism )), detailed, saturated, portrait, 50mm, F/2.8, 1m away, ((( multicolor lights )))'
@@ -56,7 +61,11 @@ def draw(data: Data):
 		generator = torch.Generator("cuda").manual_seed(seed)
 		images = pipe(prompt, negative_prompt=n_prompt, guidance_scale=7.5, generator=generator, num_images_per_prompt=1).images
 
-		grid = image_grid(images, rows=1, cols=1)
+		if 0 < int(seedno) < 5:
+			grid = images[(int(seedno) - 1)]
+		else:
+			# Limit of T4 small...
+			grid = image_grid(images, rows=1, cols=1)
 
 		fileName = "sd_" + str(time.time()) + '.png'
 		grid.save("/code/tmpdir/" + fileName)