gradio number doesn't have a type apparently

app.py

@@ -22,7 +22,7 @@ def process_image(input_image, learning_rate, iterations):
     pixel_values = pixel_values.to(device)
     pixel_values.requires_grad_(True)
 
-    for iteration in range(iterations):
+    for iteration in range(int(iterations)):
         model.zero_grad()
         if pixel_values.grad is not None:
             pixel_values.grad.data.zero_()