Update app.py

app.py

@@ -7,8 +7,8 @@ import spaces
 
 device=torch.device('cuda')
 
-
-
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.to("cuda")
  
 
 
@@ -33,7 +33,7 @@ def generate_sketch(prompt,style, num_inference_steps, guidance_scale):
     #        return gr.update(visible=False),gr.Text(value="Inappropriate prompt detected. Please try another prompt.")
     print(prompt)
     
-    pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+
     if style=='shou_xin':
         prompt= "shou_xin, " + prompt
         pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
@@ -42,7 +42,7 @@ def generate_sketch(prompt,style, num_inference_steps, guidance_scale):
         pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
     
     pipe.fuse_lora(lora_scale=1.5)
-    pipe.to("cuda")
+    
         
     image = pipe("sketched style, " + prompt, 
                  num_inference_steps=num_inference_steps,