unnati026 commited on
Commit
b25b6d4
1 Parent(s): d16d25d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- 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.')
112
 
113
  if uploaded_file is not None:
114
  # Load and preprocess the uploaded image
115
- input_image = load_image(uploaded_file, max_dim=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