Update app.py

app.py

@@ -1,33 +1,24 @@
-# Add these imports FIRST
-import sys
-import warnings
-warnings.filterwarnings("ignore", message=".*cached_download.*")
-
-# Workaround for huggingface_hub compatibility
-from huggingface_hub import cached_download as hf_hub_download
-sys.modules['huggingface_hub'].cached_download = hf_hub_download
-
-# Now import other libraries
-from diffusers import StableDiffusionPipeline
-import torch
-import random
+# Replace ALL code with this updated version
 from diffusers import StableDiffusionPipeline
 import torch
 import random
 
 def generate_image(prompt):
-    # Use updated model loading syntax
     pipe = StableDiffusionPipeline.from_pretrained(
         "stabilityai/stable-diffusion-2-1",
-        torch_dtype=torch.float16,  # Add this for better performance
-        use_auth_token="YOUR_HF_TOKEN",  # Replace with your token
-        safety_checker=None  # Optional: disable safety filter
+        torch_dtype=torch.float16,
+        use_safetensors=True,  # Add this
+        variant="fp16"         # Add this
     )
     
-    # Generate random seed
     seed = random.randint(0, 1000000)
     generator = torch.Generator().manual_seed(seed)
     
-    # Generate image
-    image = pipe(prompt, generator=generator).images[0]
-    return image
+    image = pipe(
+        prompt,
+        generator=generator,
+        num_inference_steps=25
+    ).images[0]
+    return image
+
+# Rest of your Gradio code remains the same