Update app.py

app.py

@@ -10,8 +10,6 @@ import spaces
 from numpy import uint8, rot90, fliplr
 from monai.transforms import Rotate90
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
 image_base64 = utils.image_to_base64("anatomy_aware_pipeline.png")
 article_html = f"<img src='data:image/png;base64,{image_base64}' alt='Anatomical pipeline illustration' style='width:100%;'>"
 
@@ -53,6 +51,7 @@ footer {
 """
 @spaces.GPU(duration=20)
 def predict_image(input_image, input_file):
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     if input_image is not None:
         image_path = input_image