Update app.py

app.py

@@ -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 = col1.radio('Number of qubits (should match SRV)', [3,4,5,6,7,8], index=2)
+num_of_qubits = col2.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 col1.button('Generate circuits'):    
+if col2.button('Generate circuits'):    
     image = get_correct_qcs_image(ast.literal_eval(srv), num_of_qubits, max_gates, g)
-    col2.image(image, use_column_width=False)
+    st.image(image, use_column_width=False)