Update app.py

app.py

@@ -1,30 +1,27 @@
-
+# ==============================
 # Basic Interface
+# ==============================
 
 #import gradio as gr
-
 #demo = gr.load("mswhite/miamivice", src="models")  
 #demo.launch()
 
+# ==============================
 # Detailed Interface
+# ==============================
 
 import os
 import gradio as gr
-#from transformers import pipeline
-#from diffusers import StableDiffusionImg2ImgPipeline
-
 import torch
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 
 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"
-#image = pipe(prompt).images[0]
-#pipe = pipeline("text-to-image", model="mswhite/miamivice")
+# Sample Prompt - "bright clean clear illustration side profile of a woman wearing pink lipstick p_nagel"
 
 def nagelize(text):
-    prompt = "p_nagel illustration of " + text
+    prompt = "bright clean clear illustration of " + text + " p_nagel"
     print(prompt)
     return pipe(prompt).images[0]
 
@@ -46,10 +43,3 @@ with gr.Blocks() as demo:
 demo.launch()
 
 
-
-
-
-
-
-
-