Update app.py
Browse files
app.py
CHANGED
@@ -20,10 +20,7 @@ device="cpu"
|
|
20 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
|
21 |
pipe.to(device)
|
22 |
|
23 |
-
source_img = gr.Image(
|
24 |
-
source="upload",
|
25 |
-
type="filepath")
|
26 |
-
|
27 |
gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
|
28 |
|
29 |
def infer(prompt, init_image):
|
|
|
20 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
|
21 |
pipe.to(device)
|
22 |
|
23 |
+
source_img = gr.Image(source="upload")
|
|
|
|
|
|
|
24 |
gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
|
25 |
|
26 |
def infer(prompt, init_image):
|