kfahn commited on
Commit
48c7266
1 Parent(s): a15e3f7

Update app.py

Browse files

still trouble shooting

Files changed (1) hide show
  1. app.py +3 -4
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 = [prompts, negative_prompts, conditioning_image], outputs = "gallery")
 
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]])