Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
|
19 |
pipe = pipe.to(device)
|
20 |
|
21 |
MAX_SEED = np.iinfo(np.int32).max
|
22 |
-
MAX_IMAGE_SIZE =
|
23 |
|
24 |
from datasets import load_dataset, Dataset
|
25 |
|
@@ -131,7 +131,7 @@ with gr.Blocks(css=css) as demo:
|
|
131 |
minimum=0,
|
132 |
maximum=MAX_IMAGE_SIZE,
|
133 |
step=32,
|
134 |
-
value=
|
135 |
)
|
136 |
|
137 |
height = gr.Slider(
|
@@ -139,7 +139,7 @@ with gr.Blocks(css=css) as demo:
|
|
139 |
minimum=0,
|
140 |
maximum=MAX_IMAGE_SIZE,
|
141 |
step=32,
|
142 |
-
value=
|
143 |
)
|
144 |
|
145 |
with gr.Row():
|
|
|
19 |
pipe = pipe.to(device)
|
20 |
|
21 |
MAX_SEED = np.iinfo(np.int32).max
|
22 |
+
MAX_IMAGE_SIZE = 724
|
23 |
|
24 |
from datasets import load_dataset, Dataset
|
25 |
|
|
|
131 |
minimum=0,
|
132 |
maximum=MAX_IMAGE_SIZE,
|
133 |
step=32,
|
134 |
+
value=624, # Replace with defaults that work for your model
|
135 |
)
|
136 |
|
137 |
height = gr.Slider(
|
|
|
139 |
minimum=0,
|
140 |
maximum=MAX_IMAGE_SIZE,
|
141 |
step=32,
|
142 |
+
value=724, # Replace with defaults that work for your model
|
143 |
)
|
144 |
|
145 |
with gr.Row():
|