Update

README.md

@@ -4,7 +4,7 @@ emoji: 🌍
 colorFrom: gray
 colorTo: purple
 sdk: gradio
-sdk_version: 3.45.2
+sdk_version: 4.8.0
 app_file: app.py
 license: mit
 pinned: false

app.py

@@ -8,10 +8,11 @@ import random
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from diffusers import AutoencoderKL, DiffusionPipeline
 
-DESCRIPTION = "# SD-XL"
+DESCRIPTION = "# SDXL"
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
@@ -62,6 +63,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
+@spaces.GPU
 def generate(
     prompt: str,
     negative_prompt: str = "",

requirements.txt

@@ -1,7 +1,8 @@
-accelerate==0.23.0
-diffusers==0.21.3
-gradio==3.45.2
+accelerate==0.25.0
+diffusers==0.24.0
+gradio==4.8.0
 invisible-watermark==0.2.0
-Pillow==10.0.1
-torch==2.0.1
-transformers==4.33.3
+Pillow==10.1.0
+spaces==0.18.0
+torch==2.0.0
+transformers==4.35.2