Update app.py

app.py

@@ -16,12 +16,14 @@ def main():
 
     # File uploader
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-    size = st.selectbox('Select Size',('3X4','3.5X4.5'))
+    size = st.selectbox('Select Size',('3X4','3.5X4.5','2.0X2.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)
+    else:
+        num_img = st.slider('Number of Image', 0, 100, 10)
 
     if uploaded_file is not None:
         if st.button('Make'):