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]])
|