Update app.py

app.py

@@ -109,10 +109,11 @@ for i in range(1, 11):
 epochs = st.sidebar.slider("Epochs", 1, 5, 2, help="Number of training epochs")
 steps_per_epoch = st.sidebar.slider("Steps per Epoch", 1, 100, 50, help="Number of steps per epoch")
 weight = st.sidebar.slider("Weight", 0.01, 0.1, 0.02, step=0.01, help="Weight for gradient ascent")
+max_dim = st.sidebar.slider("Image Size", 128, 1024, 300, help='Choose maximum dimension of image.\nNote: Higher size will take longer to be generated.')
 
 if uploaded_file is not None:
     # Load and preprocess the uploaded image
-    input_image = load_image(uploaded_file, max_dim=250)
+    input_image = load_image(uploaded_file, max_dim=max_dim)
     preprocessed_image = inception_v3.preprocess_input(input_image)
 
     # Create Inception model and modify for deep dream