Update app.py

app.py

@@ -1,26 +1,31 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline
-import os
+#from diffusers import StableDiffusionPipeline
+
+
+from diffusers import DiffusionPipeline
+
 
-# Retrieve the Hugging Face token stored in Hugging Face Spaces secrets
-HUGGINGFACE_TOKEN = os.getenv("keyss")
 
-if not HUGGINGFACE_TOKEN:
-    raise ValueError("Hugging Face token not found! Make sure it's added in Hugging Face Secrets.")
 
 def image_generation(prompt):
     # Check if GPU is available
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     # Load the Stable Diffusion 3 pipeline
-    pipeline = StableDiffusionPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-3-medium-diffusers",
+    pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1",
         torch_dtype=torch.float16 if device == "cuda" else torch.float32,
         use_auth_token=HUGGINGFACE_TOKEN,  # Use the Hugging Face token for authentication
         text_encoder_3=None,
-        tokenizer_3=None
-    )
+        tokenizer_3=None)
+
+    #pipeline = StableDiffusionPipeline.from_pretrained(
+       # "stabilityai/stable-diffusion-3-medium-diffusers",
+       # torch_dtype=torch.float16 if device == "cuda" else torch.float32,
+       # use_auth_token=HUGGINGFACE_TOKEN,  # Use the Hugging Face token for authentication
+       # text_encoder_3=None,
+       # tokenizer_3=None
+   # )
     
     # Enable efficient model execution
     pipeline.enable_model_cpu_offload()