mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-01-12 10:34:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

integration with W&B

Browse Source
This commit is contained in:
Abubakar Abid 2021-06-23 14:43:51 -05:00
parent ac796a15e5
commit 52d49a63f4
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
gradio/interface.py
View File

@ -548,15 +548,22 @@ class Interface:
return app, path_to_local_server, share_url
def integrate(self, comet_ml=None):
def integrate(self, comet_ml=None, wandb=None):
if comet_ml is not None:
comet_ml.log_other("Created from", "Gradio")
if self.share_url is not None:
print("")
comet_ml.log_text("gradio: " + self.share_url)
comet_ml.end()
else:
comet_ml.log_text("gradio: " + self.local_url)
comet_ml.end()
if wandb is not None:
if self.share_url is not None:
wandb.log({"Gradio panel": wandb.Html('<iframe src="' + self.share_url + '" width="' + str(self.width) + '" height="' + str(self.height) + '" frameBorder="0"></iframe>')})
else:
print("The WandB integration requires you to `launch(share=True)` first.")
def show_tip(io):