Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -684,7 +684,7 @@ def zero_inference_api(prompt, radio="model-v2", preset=PRESET_Q, h=1216, w=832,
|
|
| 684 |
image.save(tmpfile, "png", pnginfo=metadata)
|
| 685 |
return tmpfile.name, seed
|
| 686 |
|
| 687 |
-
pipe, pipe2, pipe_img2img, pipe2_img2img =
|
| 688 |
pipe, pipe2 = pipe.to(device), pipe2.to(device)
|
| 689 |
|
| 690 |
with gr.Blocks(theme=theme) as demo:
|
|
|
|
| 684 |
image.save(tmpfile, "png", pnginfo=metadata)
|
| 685 |
return tmpfile.name, seed
|
| 686 |
|
| 687 |
+
pipe, pipe2, pipe_img2img, pipe2_img2img = get_pipeline_initialize()
|
| 688 |
pipe, pipe2 = pipe.to(device), pipe2.to(device)
|
| 689 |
|
| 690 |
with gr.Blocks(theme=theme) as demo:
|