Update app.py

app.py

@@ -11,7 +11,7 @@ import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
 if not torch.cuda.is_available():
-    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
+    DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"
@@ -54,7 +54,7 @@ def generate(
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)   
 
-    options = {
+    image= pipe(
         "prompt":prompt,
         "negative_prompt":negative_prompt,
         "width":width,
@@ -64,12 +64,9 @@ def generate(
         "generator":generator,
         "use_resolution_binning":use_resolution_binning,
         "output_type":"pil",
+    ).images[0]
 
-    }
-    
-    images = pipe(**options).images[0]
-
-    return images, seed
+    return image, seed
 
 
 examples = [