Update app.py

app.py

@@ -118,8 +118,17 @@ if uploaded_file is not None:
     selected_layers = [f'mixed{i}' for i, checkbox in enumerate(layer_checkboxes, start=1) if checkbox]
     dream_model = deep_dream_model(inception, selected_layers)
 
-    # Run gradient ascent
-    image_array = run_gradient_ascent(dream_model, preprocessed_image, epochs=epochs, steps_per_epoch=steps_per_epoch, weight=weight)
+    # Run gradient ascent with progress bar
+    progress_bar = st.progress(0.0)  # Initialize progress bar
+    for epoch in range(epochs):
+        for step in range(steps_per_epoch):
+            _, grads = get_loss_and_gradient(dream_model, preprocessed_image, total_variation_weight)
+            preprocessed_image = preprocessed_image + grads * weight
+            preprocessed_image = tf.clip_by_value(preprocessed_image, -1.0, 1.0)
+            
+            # Update progress bar
+            progress_value = ((epoch * steps_per_epoch) + step) / (epochs * steps_per_epoch)
+            progress_bar.progress(progress_value)
 
     # Convert numpy arrays to PIL images
     dream_pil_image = array_to_img(deprocess_inception_image(image_array))