Update app.py
Browse files
app.py
CHANGED
@@ -108,11 +108,11 @@ for i in range(1, 11):
|
|
108 |
epochs = st.sidebar.slider("Epochs", 1, 5, 2, help="Number of training epochs")
|
109 |
steps_per_epoch = st.sidebar.slider("Steps per Epoch", 1, 100, 50, help="Number of steps per epoch")
|
110 |
weight = st.sidebar.slider("Weight", 0.01, 0.1, 0.02, step=0.01, help="Weight for gradient ascent")
|
111 |
-
|
112 |
|
113 |
if uploaded_file is not None:
|
114 |
# Load and preprocess the uploaded image
|
115 |
-
input_image = load_image(uploaded_file, max_dim=
|
116 |
preprocessed_image = inception_v3.preprocess_input(input_image)
|
117 |
|
118 |
# Create Inception model and modify for deep dream
|
|
|
108 |
epochs = st.sidebar.slider("Epochs", 1, 5, 2, help="Number of training epochs")
|
109 |
steps_per_epoch = st.sidebar.slider("Steps per Epoch", 1, 100, 50, help="Number of steps per epoch")
|
110 |
weight = st.sidebar.slider("Weight", 0.01, 0.1, 0.02, step=0.01, help="Weight for gradient ascent")
|
111 |
+
dim = st.sidebar.slider("Image Size", 128, 1024, 300, help='Choose maximum dimension of image. Higher size image will take longer to be generated.')
|
112 |
|
113 |
if uploaded_file is not None:
|
114 |
# Load and preprocess the uploaded image
|
115 |
+
input_image = load_image(uploaded_file, max_dim=dim)
|
116 |
preprocessed_image = inception_v3.preprocess_input(input_image)
|
117 |
|
118 |
# Create Inception model and modify for deep dream
|