mgyigit commited on
Commit
02d2df4
1 Parent(s): 4902a21

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +5 -2
gradio_app.py CHANGED
@@ -53,7 +53,10 @@ def function(model_name: str, num_molecules: int, seed_num: int) -> tuple[PIL.Im
53
  config.sample_num = num_molecules
54
 
55
  if seed_num is not None:
56
- config.seed = seed_num
 
 
 
57
  else:
58
  config.seed = random.randint(0, 10000)
59
 
@@ -130,7 +133,7 @@ with gr.Blocks() as demo:
130
  maximum=10_000,
131
  )
132
 
133
- seed_num = gr.Number(
134
  label="RNG seed value (can be used for reproducibility):"
135
  )
136
 
 
53
  config.sample_num = num_molecules
54
 
55
  if seed_num is not None:
56
+ try:
57
+ config.seed = int(seed_num)
58
+ except:
59
+ return
60
  else:
61
  config.seed = random.randint(0, 10000)
62
 
 
133
  maximum=10_000,
134
  )
135
 
136
+ seed_num = gr.Textbox(
137
  label="RNG seed value (can be used for reproducibility):"
138
  )
139