um235 commited on
Commit
3ab18e3
·
verified ·
1 Parent(s): e17691d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,7 +16,7 @@ else:
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
  #pipe.unet.load_adapter("um235/cartoon_cat_stickers")
19
- #pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
20
  pipe = pipe.to(device)
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
@@ -59,7 +59,7 @@ def infer(
59
 
60
 
61
  examples = [
62
- "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
63
  "An astronaut riding a green horse",
64
  "A delicious ceviche cheesecake slice",
65
  ]
@@ -121,7 +121,7 @@ with gr.Blocks(css=css) as demo:
121
  minimum=256,
122
  maximum=MAX_IMAGE_SIZE,
123
  step=32,
124
- value=1024, # Replace with defaults that work for your model
125
  )
126
 
127
  height = gr.Slider(
@@ -129,7 +129,7 @@ with gr.Blocks(css=css) as demo:
129
  minimum=256,
130
  maximum=MAX_IMAGE_SIZE,
131
  step=32,
132
- value=1024, # Replace with defaults that work for your model
133
  )
134
 
135
  with gr.Row():
 
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
  #pipe.unet.load_adapter("um235/cartoon_cat_stickers")
19
+ pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
20
  pipe = pipe.to(device)
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
 
59
 
60
 
61
  examples = [
62
+ "Sticker, cartoon-style cat character with soft yellow fur. A gentle cat with expressive eyes that shine with a sad, emotional look. The cat, with a small pink nose and a flower on its head, appears to be crying, with blue teardrops around its eyes, giving the sticker a simple yet poignant design.",
63
  "An astronaut riding a green horse",
64
  "A delicious ceviche cheesecake slice",
65
  ]
 
121
  minimum=256,
122
  maximum=MAX_IMAGE_SIZE,
123
  step=32,
124
+ value=512, # Replace with defaults that work for your model
125
  )
126
 
127
  height = gr.Slider(
 
129
  minimum=256,
130
  maximum=MAX_IMAGE_SIZE,
131
  step=32,
132
+ value=512, # Replace with defaults that work for your model
133
  )
134
 
135
  with gr.Row():