Upload 3 files

app.py

@@ -0,0 +1,20 @@
+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
+    )
+    
+    # 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

dockerfile

@@ -0,0 +1,5 @@
+FROM python:3.9-slim  # Use Python 3.9 instead of 3.10
+WORKDIR /app
+COPY . .
+RUN pip install -r requirements.txt
+CMD ["python", "app.py"]

requirements.txt

@@ -0,0 +1,5 @@
+gradio>=3.50.0
+diffusers>=0.24.0
+torch>=2.1.0
+huggingface_hub==0.19.4  # Explicit version with cached_download
+accelerate>=0.26.0