dn6 HF staff commited on
Commit
db0d363
1 Parent(s): 6860fc5

change how experiment is started

Browse files
Files changed (1) hide show
  1. app.py +15 -8
app.py CHANGED
@@ -23,7 +23,7 @@ Then use the path to a Space to generate from in the Diffusion Model tab
23
  """
24
 
25
 
26
- def create_experiment(
27
  comet_api_key,
28
  comet_workspace,
29
  comet_project_name,
@@ -43,12 +43,19 @@ def create_experiment(
43
  )
44
 
45
  try:
46
- api_experiment = comet_ml.APIExperiment(
47
- api_key=comet_api_key,
48
- workspace=comet_workspace,
49
- project_name=comet_project_name,
50
- experiment_name=comet_experiment_name,
51
- )
 
 
 
 
 
 
 
52
  experiment = {
53
  "api_key": comet_api_key,
54
  "workspace": comet_workspace,
@@ -152,7 +159,7 @@ def start_interface():
152
  experiment_state = gr.Variable(label="Experiment State")
153
 
154
  start.click(
155
- create_experiment,
156
  inputs=[
157
  comet_api_key,
158
  comet_workspace,
 
23
  """
24
 
25
 
26
+ def start_experiment(
27
  comet_api_key,
28
  comet_workspace,
29
  comet_project_name,
 
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(
54
+ api_key=comet_api_key,
55
+ workspace=comet_workspace,
56
+ project_name=comet_project_name,
57
+ experiment_name=comet_experiment_name,
58
+ )
59
  experiment = {
60
  "api_key": comet_api_key,
61
  "workspace": comet_workspace,
 
159
  experiment_state = gr.Variable(label="Experiment State")
160
 
161
  start.click(
162
+ start_experiment,
163
  inputs=[
164
  comet_api_key,
165
  comet_workspace,