Update app.py

app.py

@@ -1,19 +1,29 @@
 import os
 import gradio as gr
 import torch
-from diffusers import StableDiffusionXLPipeline, AutoencoderKL, KDPM2AncestralDiscreteScheduler
+from diffusers import StableDiffusion3Pipeline, AutoencoderKL, KDPM2AncestralDiscreteScheduler
 from huggingface_hub import hf_hub_download
 import spaces
 from PIL import Image
 import requests
 from translatepy import Translator
 
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 translator = Translator()
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 # Constants
 model = "stabilityai/stable-diffusion-3-medium"
 vae_model = "madebyollin/sdxl-vae-fp16-fix"
 
+
+model_path = hf_hub_download(
+    repo_id="stabilityai/stable-diffusion-3-medium", 
+    ignore_patterns=["*.md", "*..gitattributes"],
+    local_dir="model",
+    token=HF_TOKEN,
+    )
+
+
 CSS = """
 .gradio-container {
   max-width: 690px !important;
@@ -39,7 +49,7 @@ vae = AutoencoderKL.from_pretrained(
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusion3Pipeline.from_pretrained(model_path, vae=vae, torch_dtype=torch.float16).to("cuda")
 
 
 # Function