Update app.py

app.py

@@ -8,6 +8,7 @@
 
 # Detailed Interface
 
+import os
 import gradio as gr
 #from transformers import pipeline
 #from diffusers import StableDiffusionImg2ImgPipeline
@@ -16,14 +17,16 @@ import torch
 from diffusers import StableDiffusionPipeline
 
 pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice") #, torch_dtype=torch.float16)
-#pipe = pipe.to("cuda")
+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"]
+    prompt = "p_nagel illustration of " + text
+    print(prompt)
+    return pipe(prompt)[0]["image"]
 
 def clear(text):    
     return gr.Image.update()
@@ -31,7 +34,8 @@ def clear(text):
 with gr.Blocks() as demo:
     
     with gr.Row():
-        gr.Markdown("<img src=https://huggingface.co/spaces/mswhite/miamivice/resolve/main/miami.jpg width=720px>")
+        logo_image    = gr.Image(label = None, value = os.path.join(os.path.dirname(__file__), "miami.jpg")).style(width=720)
+        #gr.Markdown("<img src=https://huggingface.co/spaces/mswhite/miamivice/resolve/main/miami.jpg width=720px>")
         output        = gr.Image(label="Output").style(width=720)
     with gr.Row():
         art_to_draw   = gr.Textbox(label="Art to Draw")