Update app.py

app.py

@@ -3,15 +3,15 @@ import gradio as gr
 import torch
 from PIL import Image
 
-model_id = 'Dunkindont/Foto-Assisted-Diffusion-FAD_V0'
-prefix = ''
-     
-scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="FAD768-Diffuser Model/scheduler")
+model_id = 'wavymulder/portraitplus'
+prefix = 'portrait+ style,'
+
+scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
 
 pipe = StableDiffusionPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-    scheduler=scheduler)
+  scheduler=scheduler)
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,
@@ -51,7 +51,7 @@ def _parse_args(prompt, generator):
                     ), f"Config override '{var_val}' does not have the form 'VAR=val'"
                     conf_args.add_opt(opt, var_val[0], var_val[1], force_override=True)
 
-def inference(prompt, guidance, steps, width=768, height=768, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
+def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
   generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
   prompt = f"{prefix} {prompt}" if auto_prefix else prompt
 
@@ -107,37 +107,33 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="main-div">
               <div>
-                <h1 style="color:purple; font-family: verdana;">Foto Assisted Diffusion (FAD) </h1>
+                <h1 style="color:yellow;">📸 Portrait Plus 📸</h1>
               </div>
               <p>
-               Demo for <a href="https://huggingface.co/Dunkindont"><abbr title="Foto Assisted Diffusion">(FAD)</abbr></a>
-               Stable Diffusion model by <a href="https://huggingface.co/Dunkindont"><abbr title="Dunkindont">Dunkindont</abbr></a>.  {"" if prefix else ""}
-              Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else f"<b>CPU 🔥</b>"}. </p>
+               Demo for <a href="https://huggingface.com/wavymulder/portraitplus">Open Journey V4</a> 
+               Stable Diffusion model by <a href="https://huggingface.co/wavymulder/"><abbr title="Wavymulder">Wavymulder</abbr></a>. {"" if prefix else ""}  
+              Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else f"<b>CPU ⚡</b>"}. 
+               </p>
+             <p>Please use the prompt template below to achieve the desired result:
+               </p>
                
-               <blockquote>
-                • Model is meant to mimic a modern HDR photography style. 
-                • Trained on 600 HDR images on SD1.5
-                • Merged with one of my own models for illustrations and drawings for miniscule uses outside of photography
-                • No hi-res fix required, can generate natively from supported resolutions (See Supported Resolutions Tab)
-                </blockquote>
-           Please use this prompt template below to get an example of the desired generation results:
-                <br>
-<q><em>Important note: You can use the model at 512x512 but the results will likely be undesirable.</em></q>
-<br>
 <b>Prompt</b>:
-<small><code>
-medium portrait (close up:0.5) of a Beautiful scottish ginger woman, (messy bun), sitting on a wooden chair, wearing a fancy Victorian era dress, seductively posing, pale skin, dark red lips, dark eye shadow, ornate pendant, sitting in front of a large stone fireplace, candle burning, in a luxurious fantasy castle, side lighting, cinematic, Renaissance style, (Fujifilm XT3:1.1), (high detailed face:1.3), perfect hands
-</code></small>
-
-<b>Negative Prompt</b>:
-<code>
-Negative prompt: (deformed mouth), (deformed lips), (deformed eyes), (cross-eyed), (deformed iris), (deformed hands), lowers, 3d render, cartoon, long body, wide hips, narrow waist, disfigured, ugly, cross eyed, squinting, grain, Deformed, blurry, bad anatomy, poorly drawn face, mutation, mutated, extra limb, ugly, (poorly drawn hands), missing limb, floating limbs, disconnected limbs, malformed hands, blur, out of focus, long neck, disgusting, poorly drawn, mutilated, , mangled, old, surreal, ((text))	
-</code>
+<details><code>
+portrait+ style, <b>* the subject * </b>, (high detailed skin:1.2), 8k uhd, dslr, soft lighting, high quality, film grain, realistic, photo-realistic, full length frame, High detail RAW color art, piercing, diffused soft lighting, shallow depth of field, sharp focus, hyperrealism, cinematic lighting
+<br>
 <br>
+<q><i>Example: full body cyborg| full-length portrait| detailed face| symmetric| steampunk| cyberpunk| cyborg| intricate detailed| to scale| hyperrealistic| cinematic lighting| digital art| concept art| mdjrny-v4 style, (high detailed skin:1.2), 8k uhd, dslr, soft lighting, high quality, film grain, Fujifilm XT3 
+</i></q>
+</code></details>
+<q><em>Important note: Portrait+ works best at a 1:1 aspect ratio, it is also successful using tall aspect ratios as well.</em></q>
+<br>
+<b>Negative Prompt</b>:
+<details><code>
+lowres, text, error, cropped, worst quality, low quality, jpeg artifacts, ugly, duplicate, morbid, mutilated, out of frame, extra fingers, mutated hands, poorly drawn hands, poorly drawn face, mutation, deformed, blurry, dehydrated, bad anatomy, bad proportions, extra limbs, cloned face, disfigured, gross proportions, malformed limbs, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, long neck, username, watermark, signature 
+</code></details>
 <br>
-Have Fun & Enjoy <a href="https://www.thafx.com"><abbr title="//THAFX">Website</abbr></a>
+Have Fun & Enjoy ⚡ <a href="https://www.thafx.com"><abbr title="Website">//THAFX</abbr></a>
 <br>
-             
             </div>
         """
     )
@@ -146,7 +142,7 @@ Have Fun & Enjoy <a href="https://www.thafx.com"><abbr title="//THAFX">Website</
         with gr.Column(scale=55):
           with gr.Group():
               with gr.Row():
-                prompt = gr.Textbox(label="Positive Prompt", show_label=False, max_lines=2, placeholder=f"{prefix} [your prompt]").style(container=False)
+                prompt = gr.Textbox(label="Prompt", show_label=False,max_lines=2,placeholder=f"{prefix} [your prompt]").style(container=False)
                 generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
 
               image_out = gr.Image(height=512)
@@ -156,7 +152,7 @@ Have Fun & Enjoy <a href="https://www.thafx.com"><abbr title="//THAFX">Website</
           with gr.Tab("Options"):
             with gr.Group():
               neg_prompt = gr.Textbox(label="Negative prompt", placeholder="What to exclude from the image")
-              auto_prefix = gr.Checkbox(label="Prefix styling tokens automatically (RAW photo,)", value=prefix, visible=prefix)
+              auto_prefix = gr.Checkbox(label="Prefix styling tokens automatically (mdjrny-v4 style,)", value=prefix, visible=prefix)
 
               with gr.Row():
                 guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)
@@ -183,4 +179,4 @@ Have Fun & Enjoy <a href="https://www.thafx.com"><abbr title="//THAFX">Website</
     
 
 demo.queue(concurrency_count=1)
-demo.launch()
+demo.launch()