Spaces:
Sleeping
Sleeping
Removed source parameter
Browse filesRemoved source parameter since it is not accepted in the current Gradio version.
app.py
CHANGED
|
@@ -11,7 +11,7 @@ controlnet = ControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.fl
|
|
| 11 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 12 |
model_id,
|
| 13 |
controlnet=controlnet,
|
| 14 |
-
safety_checker=None,
|
| 15 |
torch_dtype=torch.float32
|
| 16 |
)
|
| 17 |
pipe = pipe.to("cpu")
|
|
@@ -42,7 +42,7 @@ def generate_image(prompt, control_image, num_inference_steps=25, guidance_scale
|
|
| 42 |
)
|
| 43 |
return result.images[0]
|
| 44 |
|
| 45 |
-
|
| 46 |
with gr.Blocks() as demo:
|
| 47 |
gr.Markdown("# ControlNet Image Generator on CPU\nThis demo uses a ControlNet pipeline (openpose variant) with Stable Diffusion to generate images guided by a control image. Note: Running on CPU can be slow!")
|
| 48 |
|
|
@@ -50,7 +50,8 @@ with gr.Blocks() as demo:
|
|
| 50 |
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your image prompt here", value="A futuristic cityscape at dusk")
|
| 51 |
|
| 52 |
with gr.Row():
|
| 53 |
-
|
|
|
|
| 54 |
output_image = gr.Image(label="Generated Image", type="pil")
|
| 55 |
|
| 56 |
with gr.Row():
|
|
|
|
| 11 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 12 |
model_id,
|
| 13 |
controlnet=controlnet,
|
| 14 |
+
safety_checker=None,
|
| 15 |
torch_dtype=torch.float32
|
| 16 |
)
|
| 17 |
pipe = pipe.to("cpu")
|
|
|
|
| 42 |
)
|
| 43 |
return result.images[0]
|
| 44 |
|
| 45 |
+
|
| 46 |
with gr.Blocks() as demo:
|
| 47 |
gr.Markdown("# ControlNet Image Generator on CPU\nThis demo uses a ControlNet pipeline (openpose variant) with Stable Diffusion to generate images guided by a control image. Note: Running on CPU can be slow!")
|
| 48 |
|
|
|
|
| 50 |
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your image prompt here", value="A futuristic cityscape at dusk")
|
| 51 |
|
| 52 |
with gr.Row():
|
| 53 |
+
|
| 54 |
+
control_image_input = gr.Image(label="Control Image", type="pil")
|
| 55 |
output_image = gr.Image(label="Generated Image", type="pil")
|
| 56 |
|
| 57 |
with gr.Row():
|