Update app.py

app.py

@@ -39,7 +39,7 @@ with open('loras.json', 'r') as f:
 # Initialize the base model
 dtype = torch.bfloat16
 base_model = "shuttleai/shuttle-jaguar"
-pipe = FluxTransformer2DModel.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
 #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 torch.cuda.empty_cache()