add tv_loss to method signature

app.py

@@ -21,7 +21,7 @@ def total_variation_loss(img):
     pixel_dif2 = img[:, :, :, 1:] - img[:, :, :, :-1]
     return (torch.sum(torch.abs(pixel_dif1)) + torch.sum(torch.abs(pixel_dif2)))
 
-def process_image(input_image, learning_rate, iterations, n_targets, seed):
+def process_image(input_image, learning_rate, tv_loss, iterations, n_targets, seed):
     if input_image is None:
         return None