app.py

app.py

@@ -26,7 +26,7 @@ layer_settings = {
 step = 0.01  # Gradient ascent step size
 num_octave = 3  # Number of scales at which to run gradient ascent
 octave_scale = 1.4  # Size ratio between scales
-iterations = 20  # Number of ascent steps per scale
+#iterations = 20  # Number of ascent steps per scale
 max_loss = 15.0
 
 def preprocess_image(img):
@@ -95,7 +95,7 @@ def gradient_ascent_loop(img, iterations, learning_rate, max_loss=None):
     return img
     
     
-def process_image(img):
+def process_image(img,iterations):
   original_img = preprocess_image(img)
   original_shape = original_img.shape[1:3]
 
@@ -123,9 +123,10 @@ def process_image(img):
   return deprocess_image(img.numpy())
   
 image = gr.inputs.Image()
+slider = gr.inputs.Slider(minimum=5, maximum=30, step=1, default=20, label="Number of ascent steps per scale")
 label = gr.outputs.Image()
 
-iface = gr.Interface(process_image,image,label,
+iface = gr.Interface(process_image,[image,slider],label,
 	#outputs=[
 	#        gr.outputs.Textbox(label="Engine issue"),
        	#	gr.outputs.Textbox(label="Engine issue score")],