ritwikraha commited on
Commit
2257008
·
1 Parent(s): 913b2fe
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,12 +41,13 @@ def generate_sketch(prompt, negative_prompt="ugly face, multiple bodies, bad ana
41
 
42
  # Gradio Interface
43
  description = """
44
- This demo utilizes the SDXL model LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were trained on sketches of Khabib by ritwikraha using DreamBooth.
45
  """
46
  # Setup Gradio interface
47
  with gr.Blocks() as demo:
48
  gr.HTML("<h1><center>Khabib Sketch Maker 🥋</center></h1>")
49
  gr.Markdown(description)
 
50
  with gr.Row():
51
  with gr.Column():
52
  prompt_input = gr.Textbox(label="Enter your image prompt", value="a sketch of TOK khabib dancing, monchrome, pen sketch", scale=8)
 
41
 
42
  # Gradio Interface
43
  description = """
44
+ This demo utilizes the SDXL model LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were trained on sketches of Khabib by ritwikraha using DreamBooth and can be found here: https://huggingface.co/ritwikraha/khabib_sketch_LoRA
45
  """
46
  # Setup Gradio interface
47
  with gr.Blocks() as demo:
48
  gr.HTML("<h1><center>Khabib Sketch Maker 🥋</center></h1>")
49
  gr.Markdown(description)
50
+ gr.HTML("<ul><li>Remember to prompt with the identifier 'TOK', e.g., A sketch of TOK Khabib.</li><li>Sketches work best.</li><li>Lower guidance scale is preferred for simpler prompts.</li><li>Negative prompt needs specifying for good image generation.</li></ul>")
51
  with gr.Row():
52
  with gr.Column():
53
  prompt_input = gr.Textbox(label="Enter your image prompt", value="a sketch of TOK khabib dancing, monchrome, pen sketch", scale=8)