Update app.py

app.py

@@ -7,6 +7,7 @@ from diffusers import StableDiffusion3Pipeline, AutoencoderKL, SD3Transformer2DM
 import spaces
 from PIL import Image
 import requests
+import transformers
 from translatepy import Translator
 
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -40,11 +41,21 @@ vae = AutoencoderKL.from_pretrained(
     torch_dtype=torch.float16,
 )
 
+transformer = SD3Transformer2DModel.from_pretrained(
+    repo, 
+    subfolder="transformer",
+    torch_dtype=torch.float16,
+)
 
+text_encoder_3 = T5EncoderModel.from_pretrained(
+    repo, 
+    subfolder="text_encoder_3", 
+    torch_dtype=torch.float16,
+)
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(repo, vae=vae, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusion3Pipeline.from_pretrained(repo, vae=vae, transformer=transformer, text_encoder_3=text_encoder_3, torch_dtype=torch.float16).to("cuda")
 
 pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(pipe.scheduler.config)