updated code to remove bug in image save
Browse files
app.py
CHANGED
|
@@ -43,6 +43,7 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
|
|
| 43 |
#global counter
|
| 44 |
#counter += 1
|
| 45 |
#if message == "revert": --to add revert functionality later
|
|
|
|
| 46 |
if counter_out > 0:
|
| 47 |
# Open the image
|
| 48 |
image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
|
|
@@ -52,7 +53,7 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
|
|
| 52 |
edited_image.save(img_name) #, overwrite=True)
|
| 53 |
else:
|
| 54 |
seed = random.randint(0, 1000000)
|
| 55 |
-
img_name = f"edited_image_{seed}.png"
|
| 56 |
edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
|
| 57 |
edited_image.save(img_name) #, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
|
| 58 |
counter_out += 1
|
|
|
|
| 43 |
#global counter
|
| 44 |
#counter += 1
|
| 45 |
#if message == "revert": --to add revert functionality later
|
| 46 |
+
print(f"counter:{counter_out}, prompt:{prompt}, img_name:{img_name}")
|
| 47 |
if counter_out > 0:
|
| 48 |
# Open the image
|
| 49 |
image_in = Image.open(img_name) #("edited_image.png") #(img_nm)
|
|
|
|
| 53 |
edited_image.save(img_name) #, overwrite=True)
|
| 54 |
else:
|
| 55 |
seed = random.randint(0, 1000000)
|
| 56 |
+
img_name = f"./edited_image_{seed}.png"
|
| 57 |
edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
|
| 58 |
edited_image.save(img_name) #, overwrite=True) #("/tmp/edited_image.png") #(img_nm)
|
| 59 |
counter_out += 1
|