Update app.py
Browse filesstill trouble shooting
app.py
CHANGED
|
@@ -23,6 +23,7 @@ def wandb_report(url):
|
|
| 23 |
return gr.HTML(iframe)
|
| 24 |
|
| 25 |
report_url = 'https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5'
|
|
|
|
| 26 |
|
| 27 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
| 28 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
|
@@ -62,9 +63,6 @@ def infer(prompts, negative_prompts, image):
|
|
| 62 |
output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
|
| 63 |
return output_images
|
| 64 |
|
| 65 |
-
|
| 66 |
-
#control_image = "https://huggingface.co/spaces/kfahn/Animal_Pose_Control_Net/blob/main/image_control.png"
|
| 67 |
-
|
| 68 |
with gr.Blocks(theme='kfahn/AnimalPose') as demo:
|
| 69 |
gr.Markdown(
|
| 70 |
"""
|
|
@@ -84,7 +82,8 @@ with gr.Blocks(theme='kfahn/AnimalPose') as demo:
|
|
| 84 |
with gr.Column():
|
| 85 |
keypoint_tool = addp5sketch(sketch_url)
|
| 86 |
#keypoint_tool = gr.HTML(lines)
|
| 87 |
-
submit_btn.click(fn=infer, inputs = [
|
|
|
|
| 88 |
|
| 89 |
#gr.Interface(fn=infer, inputs = ["text", "text", "image"], outputs = "gallery",
|
| 90 |
# examples=[["a Labrador crossing the road", "low quality", control_image]])
|
|
|
|
| 23 |
return gr.HTML(iframe)
|
| 24 |
|
| 25 |
report_url = 'https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5'
|
| 26 |
+
control_img = 'myimage.jpg'
|
| 27 |
|
| 28 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
| 29 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
|
|
|
| 63 |
output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
|
| 64 |
return output_images
|
| 65 |
|
|
|
|
|
|
|
|
|
|
| 66 |
with gr.Blocks(theme='kfahn/AnimalPose') as demo:
|
| 67 |
gr.Markdown(
|
| 68 |
"""
|
|
|
|
| 82 |
with gr.Column():
|
| 83 |
keypoint_tool = addp5sketch(sketch_url)
|
| 84 |
#keypoint_tool = gr.HTML(lines)
|
| 85 |
+
submit_btn.click(fn=infer, inputs = ["text", "text", "image"], outputs = "gallery",
|
| 86 |
+
examples=[["a Labrador crossing the road", "low quality", myimage.jpg]])
|
| 87 |
|
| 88 |
#gr.Interface(fn=infer, inputs = ["text", "text", "image"], outputs = "gallery",
|
| 89 |
# examples=[["a Labrador crossing the road", "low quality", control_image]])
|