add another size in app.py

app.py

@@ -16,7 +16,13 @@ def main():
 
     # File uploader
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-    num_img = st.slider('Number of Image', 0, 36, 12)
+    size = st.selectbox('Select Size',('3X4','3.5X4.5'))
+    size = int(''.join(i for i in size if not i in ['X','.']))
+    if size==3545:
+        num_img = st.slider('Number of Image', 0, 36, 6)
+    elif size==34:
+        num_img = st.slider('Number of Image', 0, 49, 7)
+
     if uploaded_file is not None:
         if st.button('Make'):
             # Read the image
@@ -26,7 +32,7 @@ def main():
             # st.image(image, caption='Original Image', use_column_width=True)
 
             # Convert to grayscale
-            gray_image = merge_all(num_img,img_array)
+            gray_image = merge_all(num_img,img_array,size)
             gray_image = cv2.cvtColor(np.array(gray_image), cv2.COLOR_BGR2RGB)
 
             pil_image = Image.fromarray(gray_image)