Update app.py

app.py

@@ -36,10 +36,10 @@ is_gpu_busy = False
 def get_correct_qcs_image(srv, num_of_qubits, max_gates, g):
     global is_gpu_busy
     
-    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) 
+    out_tensor           = generate_srv_tensors(pipeline, f"Generate SRV: {srv}", samples=8, 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)
 
-    fig, axs = plt.subplots(2, 3, figsize=(15,7), constrained_layout=True, dpi=200)
+    fig, axs = plt.subplots(4, 2, figsize=(7,20), constrained_layout=True, dpi=120)
     for qc,is_svr,ax in zip(qc_list, svr_list, axs.flatten()): 
         qc.draw("mpl", plot_barriers=False, ax=ax)
         ax.set_title(f"{'Correct' if is_svr==srv else 'NOT correct'}, is SRV = {is_svr}")
@@ -60,10 +60,10 @@ Generating quantum circuits with diffusion models. Official demo of [[paper-arxi
 col1, col2 = st.columns(2)
 
 srv           = col1.text_input('SRV', "[1,1,1,2,2]")
-num_of_qubits = col2.radio('Number of qubits (should match SRV)', [3,4,5,6,7,8], index=2)
+num_of_qubits = col1.radio('Number of qubits (should match SRV)', [3,4,5,6,7,8], index=2)
 max_gates     = col1.select_slider('Max gates', options=[4,8,12,16,20,24,28], value=16)
 g             = col1.slider('Guidance scale', min_value=0.0, max_value=15.0, value=7.5)
 
-if col2.button('Generate circuits'):    
+if col1.button('Generate circuits'):    
     image = get_correct_qcs_image(ast.literal_eval(srv), num_of_qubits, max_gates, g)
-    st.image(image, use_column_width=False)
+    col2.image(image, use_column_width=False)