Update app.py

app.py

@@ -2,16 +2,20 @@ import spaces
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download,login
 from transformers import pipeline
 from PIL import Image
 import os
 
+
 # Retrieve the API token from the environment variable
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 if huggingface_token is None:
     raise ValueError("HUGGINGFACE_TOKEN environment variable is not set.")
 
+# Log in to Hugging Face
+login(token=huggingface_token)    
+
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -30,7 +34,6 @@ model_path = snapshot_download(
     local_dir="stable-diffusion-3-medium",
     token=huggingface_token
 )
-#image_gen = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
 image_gen = StableDiffusion3Pipeline.from_pretrained(model_path, text_encoder_3=None, tokenizer_3=None,torch_dtype=torch.float16)
 
 image_gen = image_gen.to(device)