Update app.py

app.py

@@ -18,17 +18,21 @@ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None)
 pipe = pipe.to("cuda")
 
+pipe_img = StableDiffusionImg2ImgPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None)
+pipe_img = pipe_img.to("cuda")
+
 # Sample Prompt - "bright clean clear illustration side profile of a woman wearing pink lipstick p_nagel"
 
 def nagelize(text):
     prompt = "p_nagel detailed illustration of " + text + " bright solid colors"
-    print(prompt)
+    #print(prompt)
     return pipe(prompt).images[0]
 
-def non_nagelize(text):
-    prompt = "text"
-    print(prompt)
-    return pipe(prompt).images[0]
+def nagelize_me(source_image):
+    prompt = "p_nagel detailed illustration using bright solid colors""
+    #print(prompt)
+    #return pipe(prompt, negative_prompt=negative_prompt, image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps).images[0]
+    return pipe_img(prompt, image=source_image).images[0]
 
 with gr.Blocks() as demo:
     
@@ -38,11 +42,11 @@ with gr.Blocks() as demo:
         output        = gr.Image(label="Output").style(width=720)
     with gr.Row():
         art_to_draw   = gr.Textbox(label="Art to Draw").style(width=720)           
-        submit_btn    = gr.Button("Back to the 80s")
-        normal_btn    = gr.Button("Not going to the 80s")
+        flashb_btn    = gr.Button("Flashback to the 80s")
+        back80_btn    = gr.Button("Send me back to the 80s")
 
-    submit_btn.click(fn=nagelize, inputs=[art_to_draw], outputs=[output]) 
-    normal_btn.click(fn=non_nagelize, inputs=[art_to_draw], outputs=[output])    
+    flashb_btn.click(fn=nagelize,    inputs=[art_to_draw], outputs=[output]) 
+    back80_btn.click(fn=nagelize_me, inputs=[logo_image],  outputs=[output])    
 
 demo.launch()