Update app.py

app.py

@@ -7,14 +7,19 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
 else:
     torch_dtype = torch.float32
 
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+base_model = "black-forest-labs/FLUX.1-dev"
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+
+lora_repo = "MegaTronX/SuicideGirl-FLUX"
+trigger_word = "SuicideGirl"  # Leave trigger_word blank if not used.
+pipe.load_lora_weights(lora_repo, weight_name = "AndroFlux-v19.safetensors")
+
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max