Update app.py

app.py

@@ -7,14 +7,14 @@ 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)
+pipe = pipeline = AutoPipelineForText2Image.from_pretrained("ostris/OpenFLUX.1", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights('aleksa-codes/flux-ghibsky-illustration', weight_name='lora.safetensors')
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max