ysharma HF staff commited on
Commit
e140357
·
1 Parent(s): 490262d

update temp file problem

Browse files
Files changed (1) hide show
  1. app.py +8 -11
app.py CHANGED
@@ -5,7 +5,6 @@ import torch
5
  import gradio as gr
6
  import random
7
  from PIL import Image
8
-
9
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
10
 
11
  model_id = "timbrooks/instruct-pix2pix"
@@ -13,30 +12,28 @@ pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dt
13
  pipe.to("cuda")
14
  pipe.enable_attention_slicing()
15
 
16
- #image = PIL.Image.open("./example.png")
17
- #image = PIL.ImageOps.exif_transpose(image)
18
- #image = image.convert("RGB")
19
- #image
20
- #prompt = "turn him into cyborg"
21
- #pipe(prompt, image=image, num_inference_steps=20, image_guidance_scale=1).images[0]
22
-
23
  counter = 0
24
 
25
  def chat(image_in, message, history): #, progress=gr.Progress(track_tqdm=True)):
26
  #progress(0, desc="Starting...")
27
  global counter
 
 
28
  counter += 1
 
29
  #if message == "revert": --to add revert functionality later
30
  if counter > 1:
31
  # Open the image
32
- image_in = Image.open("./edited_image.png")
33
  prompt = message #eg - "turn him into cyborg"
34
  edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
35
- edited_image.save("edited_image.png") #("./edited_image.png")
36
  history = history or []
37
  add_text_list = ["There you go ", "Enjoy your image! ", "Nice work! Wonder what you gonna do next! ", "Way to go! ", "Does this work for you? ", "Something like this? "]
38
  #Resizing the image for better display
39
- response = random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 200px; height: 200px;">'
 
40
  history.append((message, response))
41
  return history, history
42
 
 
5
  import gradio as gr
6
  import random
7
  from PIL import Image
 
8
  from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
9
 
10
  model_id = "timbrooks/instruct-pix2pix"
 
12
  pipe.to("cuda")
13
  pipe.enable_attention_slicing()
14
 
15
+ seed = random.randint(0, 1000000)
 
 
 
 
 
 
16
  counter = 0
17
 
18
  def chat(image_in, message, history): #, progress=gr.Progress(track_tqdm=True)):
19
  #progress(0, desc="Starting...")
20
  global counter
21
+ global seed
22
+ img_nm = f"./edited_image_{seed}.png"
23
  counter += 1
24
+
25
  #if message == "revert": --to add revert functionality later
26
  if counter > 1:
27
  # Open the image
28
+ image_in = Image.open(img_nm) #("./edited_image.png")
29
  prompt = message #eg - "turn him into cyborg"
30
  edited_image = pipe(prompt, image=image_in, num_inference_steps=20, image_guidance_scale=1).images[0]
31
+ edited_image.save(img_nm) #("edited_image.png") #("./edited_image.png")
32
  history = history or []
33
  add_text_list = ["There you go ", "Enjoy your image! ", "Nice work! Wonder what you gonna do next! ", "Way to go! ", "Does this work for you? ", "Something like this? "]
34
  #Resizing the image for better display
35
+ response = random.choice(add_text_list) + '<img src="/file=' + img_nm[2:] + '" style="width: 200px; height: 200px;">'
36
+ #response = random.choice(add_text_list) + '<img src="/file=edited_image.png" style="width: 200px; height: 200px;">'
37
  history.append((message, response))
38
  return history, history
39