Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,13 +8,11 @@ Accelerator = Accelerator(cpu=True)
|
|
8 |
pipe = accelerator.prepare(UniDiffuserPipeline.from_pretrained("thu-ml/unidiffuser-v1", torch_dtype=torch.bfloat16))
|
9 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
10 |
|
11 |
-
def plex(image_url,stips)
|
12 |
-
init_image = load_image(image_url).resize((512, 512))
|
13 |
-
|
14 |
-
|
15 |
-
i2t_text =
|
16 |
-
|
17 |
-
sample = pipe(prompt=i2t_text, num_inference_steps=stips, guidance_scale=8.0)
|
18 |
for i, imge in enumerate(sample["images"]):
|
19 |
apol.append(imge)
|
20 |
return apol
|
|
|
8 |
pipe = accelerator.prepare(UniDiffuserPipeline.from_pretrained("thu-ml/unidiffuser-v1", torch_dtype=torch.bfloat16))
|
9 |
pipe = accelerator.prepare(pipe.to("cpu"))
|
10 |
|
11 |
+
def plex(image_url,stips):
|
12 |
+
init_image = load_image(image_url).resize((512, 512))
|
13 |
+
sample = pipe(image=init_image, num_inference_steps=stips, guidance_scale=8.0)
|
14 |
+
i2t_text = sample.text[0]
|
15 |
+
sample = pipe(prompt=i2t_text, num_inference_steps=stips, guidance_scale=8.0)
|
|
|
|
|
16 |
for i, imge in enumerate(sample["images"]):
|
17 |
apol.append(imge)
|
18 |
return apol
|