Update app.py

app.py

@@ -26,16 +26,16 @@ weigths_file = download(url_weights, save_destination)
 #--------------------------------
 # setup
 
+try:
+    pipeline
+except:
+    pipeline = DiffusionPipeline.from_config_file(save_destination, infer_torch_device())  
+    pipeline.scheduler.set_timesteps(20) 
+
 is_gpu_busy = False
 def get_correct_qcs_image(srv, num_of_qubits, max_gates, g):
     global is_gpu_busy
-
-    try:
-        pipeline
-    except:
-        pipeline = DiffusionPipeline.from_config_file(save_destination, infer_torch_device())  
-        pipeline.scheduler.set_timesteps(20) 
-        
+       
     out_tensor           = generate_srv_tensors(pipeline, f"Generate SRV: {srv}", samples=6, system_size=num_of_qubits, num_of_qubits=num_of_qubits, max_gates=max_gates, g=g) 
     qc_list, _, svr_list = convert_tensors_to_srvs(out_tensor, pipeline.gate_pool)