Update app.py

Adding function to and wands report

app.py

@@ -13,6 +13,11 @@ with open("test.html") as f:
 
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
+def wandb_report(url):
+    iframe = f'<iframe src ={url} style="border:none;height:1024px;width:100%"/frame>'
+    return gr.HTML(iframe)
+
+report_url = 'https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5'
 
 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
     "JFoz/dog-cat-pose", dtype=jnp.bfloat16