Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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:
|