Update app.py

app.py

@@ -9,8 +9,8 @@ pipeline.enable_model_cpu_offload() #save some VRAM by offloading the model to C
 
 
 # Set the model to the appropriate device
-#device = "cuda" if torch.cuda.is_available() else "cpu"
-#pipeline.to(device)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline.to(device)
 
 def generate_image(prompt, guidance_scale=7.5, num_inference_steps=50):
     # Generate an image based on the prompt