Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import torch
 from diffusers import StableDiffusion3Pipeline, AutoencoderKL, KDPM2AncestralDiscreteScheduler
-from huggingface_hub import hf_hub_download
+from huggingface_hub import snapshot_download
 import spaces
 from PIL import Image
 import requests
@@ -16,16 +16,17 @@ 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", 
-    filename="*.safetensors"
+model_path = snapshot_download(
+    repo_id=model, 
+    revision="refs/pr/26",
+    repo_type="model", 
+    ignore_patterns=["*.md", "*..gitattributes"],
     local_dir="model",
     token=HF_TOKEN,
     )
 
 
+
 CSS = """
 .gradio-container {
   max-width: 690px !important;