liuganghuggingface commited on
Commit
85ede68
·
verified ·
1 Parent(s): 1678cdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def save_interesting_log(smiles, properties, suggested_properties):
116
  }
117
  writer.writerow(log_data)
118
 
119
- @spaces.GPU(duration=60)
120
  def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):
121
  print('in generate graph', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
122
  model, device = model_state
 
116
  }
117
  writer.writerow(log_data)
118
 
119
+ # @spaces.GPU(duration=60)
120
  def generate_graph(CH4, CO2, H2, N2, O2, guidance_scale, num_nodes, repeating_time, model_state, num_chain_steps, fps):
121
  print('in generate graph', torch.device("cuda" if torch.cuda.is_available() else "cpu"))
122
  model, device = model_state