missing device def

app.py

@@ -70,6 +70,8 @@ def visualize(image, mask, box):
 
 # Main function for Gradio app
 def process_images(file, x_min, y_min, x_max, y_max):
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    
     # Load and preprocess image
     image, H, W = load_image(file)
     image_resized = transform.resize(image, (1024, 1024), anti_aliasing=True)