Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -435,7 +435,7 @@ def main():
|
|
| 435 |
mode = "ODE"
|
| 436 |
|
| 437 |
parser.add_argument("--num_gpus", type=int, default=1)
|
| 438 |
-
parser.add_argument("--ckpt", type=str, default="
|
| 439 |
parser.add_argument("--ema", type=bool, default=True)
|
| 440 |
parser.add_argument("--precision", default="bf16", choices=["bf16", "fp32"])
|
| 441 |
|
|
@@ -557,12 +557,12 @@ def main():
|
|
| 557 |
# ntk_scaling, proportional_attn
|
| 558 |
# ])
|
| 559 |
with gr.Column():
|
| 560 |
-
default_img = Image.open("./image.png")
|
| 561 |
output_img = gr.Image(
|
| 562 |
label="Generated image",
|
| 563 |
interactive=False,
|
| 564 |
format="png",
|
| 565 |
-
value=default_img,
|
| 566 |
)
|
| 567 |
|
| 568 |
with gr.Row():
|
|
|
|
| 435 |
mode = "ODE"
|
| 436 |
|
| 437 |
parser.add_argument("--num_gpus", type=int, default=1)
|
| 438 |
+
parser.add_argument("--ckpt", type=str, default="/home/user/app/checkpoints")
|
| 439 |
parser.add_argument("--ema", type=bool, default=True)
|
| 440 |
parser.add_argument("--precision", default="bf16", choices=["bf16", "fp32"])
|
| 441 |
|
|
|
|
| 557 |
# ntk_scaling, proportional_attn
|
| 558 |
# ])
|
| 559 |
with gr.Column():
|
| 560 |
+
# default_img = Image.open("./image.png")
|
| 561 |
output_img = gr.Image(
|
| 562 |
label="Generated image",
|
| 563 |
interactive=False,
|
| 564 |
format="png",
|
| 565 |
+
# value=default_img,
|
| 566 |
)
|
| 567 |
|
| 568 |
with gr.Row():
|