Update app.py

app.py

@@ -67,7 +67,7 @@ def get_loss_and_gradient(model, inputs, total_variation_weight=0):
     return loss, grads
 
 
-def run_gradient_ascent(model, inputs, progress_bar epochs=1, steps_per_epoch=1, weight=0.05, total_variation_weight=0):
+def run_gradient_ascent(model, inputs, progress_bar, epochs=1, steps_per_epoch=1, weight=0.05, total_variation_weight=0):
     img = tf.convert_to_tensor(inputs)
     for i in range(epochs):
         for _ in range(steps_per_epoch):
@@ -126,7 +126,7 @@ if uploaded_file is not None:
     progress_bar = st.progress(0.0)  # Initialize progress bar
     
     # Run gradient ascent
-    image_array = run_gradient_ascent(dream_model, preprocessed_image, progress_bar epochs=epochs, steps_per_epoch=steps_per_epoch, weight=weight)
+    image_array = run_gradient_ascent(dream_model, preprocessed_image, progress_bar, epochs=epochs, steps_per_epoch=steps_per_epoch, weight=weight)
 
     # Convert numpy arrays to PIL images
     dream_pil_image = array_to_img(deprocess_inception_image(image_array))