Update app.py

app.py

@@ -8,12 +8,20 @@ import random
 import spaces
 
 # Constants
-BASE_MODEL = "black-forest-labs/FLUX.1-dev"
-LORA_MODEL = "MegaTronX/SuicideGirl-FLUX"  # Replace with your LoRA path 
+#BASE_MODEL = "black-forest-labs/FLUX.1-dev"
+#LORA_MODEL = "MegaTronX/SuicideGirl-FLUX"  # Replace with your LoRA path 
 MAX_SEED = np.iinfo(np.int32).max
 
+
+
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("MegaTronX/SuicideGirl-FLUX", weight_name="SuicideGirls.safetensors")
+pipe.fuse_lora(lora_scale=0.8)
+pipe.to("cuda")
+
+
 # Initialize model and scheduler
-if torch.cuda.is_available():
+'''if torch.cuda.is_available():
     transformer = FluxTransformer2DModel.from_single_file(
         "https://huggingface.co/MegaTronX/SuicideGirl-FLUX/blob/main/SuicideGirls.safetensors",
         torch_dtype=torch.bfloat16
@@ -30,6 +38,8 @@ if torch.cuda.is_available():
     
     # Load and apply LoRA weights
     pipe.load_lora_weights(LORA_MODEL)
+'''
+
 
 @spaces.GPU
 def generate_image(