Update app.py

app.py

@@ -16,7 +16,7 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
 
-pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None)
 pipe = pipe.to("cuda")
 
 #prompt = "a photo of an astronaut riding a horse on mars"
@@ -26,7 +26,7 @@ pipe = pipe.to("cuda")
 def nagelize(text):
     prompt = "p_nagel illustration of " + text
     print(prompt)
-    return pipe(prompt)[0]["image"]
+    return pipe(prompt).images[0]
 
 def clear(text):    
     return gr.Image.update()