amazonaws-la commited on
Commit
6b90528
·
verified ·
1 Parent(s): ef90888

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -76,7 +76,6 @@ def generate(
76
  response = requests.get(url)
77
  init_image = Image.open(BytesIO(response.content)).convert("RGB")
78
  init_image = init_image.resize((1024, 1024))
79
- init_image.save('source.png')
80
 
81
  if ENABLE_CPU_OFFLOAD:
82
  pipe.enable_model_cpu_offload()
@@ -107,7 +106,6 @@ def generate(
107
  guidance_scale=guidance_scale_base,
108
  num_inference_steps=num_inference_steps_base,
109
  generator=generator,
110
- image=init_image,
111
  output_type="pil",
112
  ).images[0]
113
  else:
 
76
  response = requests.get(url)
77
  init_image = Image.open(BytesIO(response.content)).convert("RGB")
78
  init_image = init_image.resize((1024, 1024))
 
79
 
80
  if ENABLE_CPU_OFFLOAD:
81
  pipe.enable_model_cpu_offload()
 
106
  guidance_scale=guidance_scale_base,
107
  num_inference_steps=num_inference_steps_base,
108
  generator=generator,
 
109
  output_type="pil",
110
  ).images[0]
111
  else: