JoPmt commited on
Commit
dfebf05
1 Parent(s): 3e8bafc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
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
- sample = pipe(image=init_image, num_inference_steps=stips, guidance_scale=8.0)
15
- i2t_text = sample.text[0]
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