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)