modified app.py gradio cuda

app.py

@@ -91,6 +91,7 @@ import os
 # print("Preloaded images:", preloaded_images)
 @spaces.GPU
 def get_c(x, points):
+    x = utils.imagenet_normalize(x).to(device)
     with torch.no_grad():
         counterfactual = model.get_counterfactual(x, points)
     return counterfactual
@@ -257,8 +258,6 @@ with gr.Blocks() as demo:
 
             points = points[:, [1, 0, 3, 2]]
 
-            print(points)
-
             img = Image.fromarray(original_image)
 
             transform = transforms.Compose([
@@ -277,7 +276,7 @@ with gr.Blocks() as demo:
             x = img[:, :, None].expand(-1, -1, 3, -1, -1).to(torch.float16)
 
             # Imagenet-normalize the inputs (standardization)
-            x = utils.imagenet_normalize(x).to(device)
+
 
             counterfactual = get_c(x, points)