Update app.py
Browse files
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 |
-
|
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 |
-
"
|
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=
|
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=
|
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():
|