device setting defined by instance type

app.py

@@ -23,7 +23,8 @@ subprocess.call(
 )
 
 if __name__ == "__main__":
-    servicer = SdmCompressionDemo()
+    device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    servicer = SdmCompressionDemo(device)
     example_list = servicer.get_example_list()
 
     with gr.Blocks(theme='nota-ai/theme') as demo:

demo.py

@@ -11,8 +11,8 @@ DEVICE='cuda'
 # DEVICE='cpu'
 
 class SdmCompressionDemo:
-    def __init__(self) -> None:
-        self.device = DEVICE
+    def __init__(self, device) -> None:
+        self.device = device
         self.torch_dtype = torch.float16 if 'cuda' in self.device else torch.float32
         
         self.pipe_original = StableDiffusionPipeline.from_pretrained(ORIGINAL_CHECKPOINT_ID,