Update app.py

app.py

@@ -8,8 +8,8 @@ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
 # 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