kaytoo2022 commited on
Commit
623741b
·
verified ·
1 Parent(s): 7515cd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,6 +10,7 @@ dtype = torch.float16
10
 
11
  repo = "stabilityai/stable-diffusion-3-medium-diffusers"
12
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
 
13
 
14
  MAX_SEED = np.iinfo(np.int32).max
15
  MAX_IMAGE_SIZE = 1344
@@ -35,9 +36,8 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
35
  return image, seed
36
 
37
  examples = [
38
- "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
39
- "An astronaut riding a green horse",
40
- "A delicious ceviche cheesecake slice",
41
  ]
42
 
43
  css="""
 
10
 
11
  repo = "stabilityai/stable-diffusion-3-medium-diffusers"
12
  pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to(device)
13
+ pipe.load_lora_weights("kaytoo2022/cara-the-cavapoo-2")
14
 
15
  MAX_SEED = np.iinfo(np.int32).max
16
  MAX_IMAGE_SIZE = 1344
 
36
  return image, seed
37
 
38
  examples = [
39
+ "Cara the Cavapoo in a jungle, cold color palette, muted colors, detailed, 8k",
40
+ "Cara the Cavapoo riding a green horse"
 
41
  ]
42
 
43
  css="""