Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusion3Pipeline
 import os
+from huggingface_hub import snapshot_download
 
 # Retrieve the API token from the environment variable
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
@@ -11,9 +12,21 @@ if huggingface_token is None:
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+model_path = snapshot_download(
+    repo_id="stabilityai/stable-diffusion-3-medium", 
+    revision="refs/pr/26",
+    repo_type="model", 
+    ignore_patterns=["*.md", "*..gitattributes"],
+    local_dir="stable-diffusion-3-medium",
+    token=huggingface_token, # yeni bir token-id yazın.
+    )
 # Load the Stable Diffusion model
 repo = "stabilityai/stable-diffusion-3-medium-diffusers"
-image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None, use_auth_token=huggingface_token)
+#image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None, use_auth_token=huggingface_token)
+image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None)
+
+#pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+
 image_gen = image_gen.to(device)
 
 def generate_image(prompt, num_inference_steps=50, guidance_scale=7.5):