dn6 HF staff commited on
Commit
fca694c
β€’
1 Parent(s): 5894721
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -38,16 +38,19 @@ def start_experiment(
38
  Please add your API key in order to log your predictions to a Comet Experiment.
39
  If you don't have a Comet account yet, you can sign up using the link below:
40
 
41
- [Sign Up for Comet](https://www.comet.ml/signup)
42
  """,
43
  )
44
 
45
  try:
46
  if comet_experiment_name:
47
- api_experiment = comet_ml.API(api_key=comet_api_key).get_experiment(
48
- workspace=comet_workspace,
49
- project=comet_project_name,
50
- experiment_name=comet_experiment_name,
 
 
 
51
  )
52
  else:
53
  api_experiment = comet_ml.APIExperiment(
@@ -59,7 +62,7 @@ def start_experiment(
59
  "api_key": comet_api_key,
60
  "workspace": comet_workspace,
61
  "project_name": comet_project_name,
62
- "experiment_name": api_experiment.name,
63
  }
64
 
65
  return experiment, f"Started {api_experiment.name}. Happy logging!😊"
@@ -72,11 +75,11 @@ def get_experiment(experiment_state):
72
  api_key = experiment_state["api_key"]
73
  workspace = experiment_state["workspace"]
74
  project = experiment_state["project_name"]
75
- experiment_name = experiment_state["experiment_name"]
76
 
77
  try:
78
  return comet_ml.API(api_key=api_key).get_experiment(
79
- workspace=workspace, project=project, experiment_name=experiment_name
80
  )
81
  except Exception as e:
82
  return None
@@ -154,7 +157,7 @@ def start_interface():
154
  start = gr.Button("Start Experiment", variant="primary")
155
  status = gr.Button("Experiment Status")
156
 
157
- status_output = gr.Markdown(label="Status")
158
  experiment_state = gr.Variable(label="Experiment State")
159
 
160
  start.click(
 
38
  Please add your API key in order to log your predictions to a Comet Experiment.
39
  If you don't have a Comet account yet, you can sign up using the link below:
40
 
41
+ https://www.comet.ml/signup
42
  """,
43
  )
44
 
45
  try:
46
  if comet_experiment_name:
47
+ api_experiment = get_experiment(
48
+ {
49
+ "api_key": comet_api_key,
50
+ "workspace": comet_workspace,
51
+ "project_name": comet_project_name,
52
+ "experiment": comet_experiment_name,
53
+ }
54
  )
55
  else:
56
  api_experiment = comet_ml.APIExperiment(
 
62
  "api_key": comet_api_key,
63
  "workspace": comet_workspace,
64
  "project_name": comet_project_name,
65
+ "experiment": api_experiment.name,
66
  }
67
 
68
  return experiment, f"Started {api_experiment.name}. Happy logging!😊"
 
75
  api_key = experiment_state["api_key"]
76
  workspace = experiment_state["workspace"]
77
  project = experiment_state["project_name"]
78
+ experiment_name = experiment_state["experiment"]
79
 
80
  try:
81
  return comet_ml.API(api_key=api_key).get_experiment(
82
+ workspace=workspace, project_name=project, experiment=experiment_name
83
  )
84
  except Exception as e:
85
  return None
 
157
  start = gr.Button("Start Experiment", variant="primary")
158
  status = gr.Button("Experiment Status")
159
 
160
+ status_output = gr.Textbox(label="Status")
161
  experiment_state = gr.Variable(label="Experiment State")
162
 
163
  start.click(