Update app.py

app.py

@@ -5,7 +5,6 @@ import json
 
 import gradio as gr
 import numpy as np
-from PIL import Image
 import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
@@ -52,9 +51,8 @@ def generate(
 ):
     pipe.to(device)
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator().manual_seed(seed)   
-
-    image= pipe(
+    generator = torch.Generator().manual_seed(seed)
+    image = pipe(
         "prompt":prompt,
         "negative_prompt":negative_prompt,
         "width":width,
@@ -63,12 +61,9 @@ def generate(
         "num_inference_steps":num_inference_steps,
         "generator":generator,
         "use_resolution_binning":use_resolution_binning,
-        "output_type":"pil",
     ).images[0]
-
     return image, seed
 
-
 examples = [
     "a cat eating a piece of cheese",
     "a ROBOT riding a BLUE horse on Mars, photorealistic, 4k",
@@ -98,7 +93,7 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
             run_button = gr.Button("生成", scale=0)
-        result = gr.Gallery(label="返回", columns=1)
+        result = gr.Image(label="Result", show_label=False)
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
             use_negative_prompt = gr.Checkbox(label="使用反向词条", value=True)