Update app.py

app.py

@@ -120,7 +120,8 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-
-    pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=torch.float16).to("cuda")
+    # Check if CUDA is available
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=torch.float16).to(device)
     demo.queue(concurrency_count=2,
             ).launch()