Update app.py

app.py

@@ -3,16 +3,18 @@ import torch
 import random
 
 def generate_image(prompt):
-    # Load the model
+    # Use updated model loading syntax
     pipe = StableDiffusionPipeline.from_pretrained(
         "stabilityai/stable-diffusion-2-1",
-        use_auth_token="YOUR_HF_TOKEN"  # Replace with your token
+        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
     )
     
-    # Random seed for uniqueness
+    # 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
+    return image