Update app.py

app.py

@@ -18,7 +18,7 @@ for model_name in model_list:
 @spaces.GPU
 def generate_image(model_name, prompt, negative_prompt, num_inference_steps, guidance_scale):
     pipe = models[model_name]
-    image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale)["sample"][0]
+    image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale)["stabilityai/stable-diffusion-xl-base-1.0"][0]
     return image