echoece commited on
Commit
b82629b
·
verified ·
1 Parent(s): 77da420
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from diffusers import StableDiffusionPipeline
3
  import torch
4
 
5
  # ⚠️ You can replace this with any try-on model, e.g. akhaliq/TryOnDiffusion
6
- model_id = "akhaliq/TryOnDiffusion"
7
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
  pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  def try_on(person_img, cloth_img):
 
3
  import torch
4
 
5
  # ⚠️ You can replace this with any try-on model, e.g. akhaliq/TryOnDiffusion
6
+ model_id = "lllyasviel/TryOnDiffusion" # public
7
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16
8
  pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
9
 
10
  def try_on(person_img, cloth_img):