Update app.py

app.py

@@ -13,6 +13,7 @@ from huggingface_hub import hf_hub_download
 
 zero = torch.Tensor([0]).cuda()
 print(zero.device) # <-- 'cpu' 🤔
+device = zero.device # "cuda"
 
 # Gets the absolute path of the current script
 script_directory = os.path.dirname(os.path.realpath(__file__))
@@ -41,8 +42,6 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 @spaces.GPU
 def process(inputImage,prompt,negative_prompt):
 
-    device = zero.device # "cuda"
-
     # hyper-parameter
     select_images = load_image(Image.fromarray(inputImage))
     num_steps = 50