Update app.py

app.py

@@ -70,13 +70,13 @@ def get_loss_and_gradient(model, inputs, 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):
+        for step in range(steps_per_epoch):
             _, grads = get_loss_and_gradient(model, img, total_variation_weight)
             img = img + grads * weight
             img = tf.clip_by_value(img, -1.0, 1.0)
 
             # Update progress bar
-            progress_value = ((epoch * steps_per_epoch) + step) / (epochs * steps_per_epoch)
+            progress_value = ((i * steps_per_epoch) + step) / (epochs * steps_per_epoch)
             progress_bar.progress(progress_value)
 
     return img.numpy(), progress_bar