Update app.py

Adding an addp5sketch function

app.py

@@ -8,11 +8,16 @@ from PIL import Image
 from diffusers import FlaxStableDiffusionControlNetPipeline, FlaxControlNetModel
 import cv2
 
-with open("test.html") as f:
-    lines = f.readlines()
+#with open("test.html") as f:
+#    lines = f.readlines()
 
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
+
+def addp5sketch(url):
+    iframe = f'<iframe src ={url} style="border:none;height:525px;width:100%"/frame>'
+    return gr.HTML(iframe)
+
 def wandb_report(url):
     iframe = f'<iframe src ={url} style="border:none;height:1024px;width:100%"/frame>'
     return gr.HTML(iframe)
@@ -72,7 +77,7 @@ with gr.Blocks(theme='kfahn/AnimalPose') as demo:
       """)
   with gr.Column():
     with gr.Row():
-      keypoint_tool = gr.HTML(lines)
+      keypoint_tool = addp5sketch(sketch_url)
     with gr.Row():
       prompts  = gr.Textbox(label="Prompt")
     with gr.Row():