Update app.py

app.py

@@ -100,16 +100,14 @@ florence_processor = AutoProcessor.from_pretrained(LOCAL_FLORENCE, trust_remote_
 
 diffusers.StableDiffusion3Pipeline = StableDiffusionPipeline
 
-pipe = StableDiffusionPipeline(
-    text_encoder=text_encoder,
-    tokenizer=tokenizer,
-    feature_extractor=feature_extractor,
-    scheduler=scheduler,
-    unet=unet,
-    vae=vae,
-    safety_checker=safety_checker,
+pipe = DiffusionPipeline.from_pretrained(
+    "tensorart/stable-diffusion-3.5-large-TurboX",
     torch_dtype=torch.float16,
-).to(device)
+    trust_remote_code=True,
+    safety_checker=None,
+    feature_extractor=None
+)
+pipe = pipe.to("cuda")
 
 pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", local_files_only=True, trust_remote_code = True, shift=5)