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

mlflow integrate

Browse Source
This commit is contained in:
dawoodkhan82 2021-07-26 17:25:41 -04:00
parent 0d5057f6b8
commit 969782a412
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gradio/interface.py
View File

@ -593,11 +593,10 @@ class Interface:
return app, path_to_local_server, share_url
def integrate(self, comet_ml=None, wandb=None):
def integrate(self, comet_ml=None, wandb=None, mlflow=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:
@ -608,6 +607,13 @@ class Interface:
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.")
if mlflow is not None:
if self.share_url is not None:
mlflow.log_param("Gradio Interface Share Link",
self.share_url)
else:
mlflow.log_param("Gradio Interface Local Link",
self.local_url)