Update app.py

app.py

@@ -9,13 +9,24 @@
 # Detailed Interface
 
 import gradio as gr
-from transformers import pipeline
+#from transformers import pipeline
+#from diffusers import StableDiffusionImg2ImgPipeline
 
+import torch
+from diffusers import StableDiffusionPipeline
+
+pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
+
+#prompt = "a photo of an astronaut riding a horse on mars"
+#image = pipe(prompt).images[0]
 #pipe = pipeline("text-to-image", model="mswhite/miamivice")
 
 def nagelize(text):
+    return pipe("p_nagel illustration of " + text)[0]["image"]
+
+def clear(text):    
     return gr.Image.update()
-#    return pipe(text)[0]["image"]
 
 with gr.Blocks() as demo: