Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
-
model_id = 'Dunkindont/Foto-Assisted-Diffusion-FAD_V0'
|
7 |
prefix = 'RAW photo,'
|
8 |
|
9 |
scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="FAD768-Diffuser Model/scheduler")
|
@@ -51,7 +51,7 @@ def _parse_args(prompt, generator):
|
|
51 |
), f"Config override '{var_val}' does not have the form 'VAR=val'"
|
52 |
conf_args.add_opt(opt, var_val[0], var_val[1], force_override=True)
|
53 |
|
54 |
-
def inference(prompt, guidance, steps, width=
|
55 |
generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
|
56 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|
57 |
|
@@ -146,7 +146,7 @@ Have Fun & Enjoy <a href="https://www.thafx.com"><abbr title="//THAFX">Website</
|
|
146 |
with gr.Column(scale=55):
|
147 |
with gr.Group():
|
148 |
with gr.Row():
|
149 |
-
prompt = gr.Textbox(label="Prompt", show_label=False,max_lines=2,placeholder=f"{prefix} [your prompt]").style(container=False)
|
150 |
generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
|
151 |
|
152 |
image_out = gr.Image(height=512)
|
|
|
3 |
import torch
|
4 |
from PIL import Image
|
5 |
|
6 |
+
model_id = 'Dunkindont/Foto-Assisted-Diffusion-FAD_V0/FAD768-Diffuser Model'
|
7 |
prefix = 'RAW photo,'
|
8 |
|
9 |
scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="FAD768-Diffuser Model/scheduler")
|
|
|
51 |
), f"Config override '{var_val}' does not have the form 'VAR=val'"
|
52 |
conf_args.add_opt(opt, var_val[0], var_val[1], force_override=True)
|
53 |
|
54 |
+
def inference(prompt, guidance, steps, width=768, height=768, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
|
55 |
generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
|
56 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|
57 |
|
|
|
146 |
with gr.Column(scale=55):
|
147 |
with gr.Group():
|
148 |
with gr.Row():
|
149 |
+
prompt = gr.Textbox(label="Positive Prompt", placeholder="What to include in the image", show_label=False,max_lines=2,placeholder=f"{prefix} [your prompt]").style(container=False)
|
150 |
generate = gr.Button(value="Generate").style(rounded=(False, True, True, False))
|
151 |
|
152 |
image_out = gr.Image(height=512)
|