HusseinHE commited on
Commit
4b62a43
1 Parent(s): 4960dc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -18,7 +18,9 @@ from share_btn import community_icon_html, loading_icon_html, share_js
18
  from gallery_history import fetch_gallery_history, show_gallery_history
19
  from illusion_style import css
20
 
21
- BASE_MODEL = "HusseinHE/icbinh"
 
 
22
 
23
  # Initialize both pipelines
24
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
@@ -148,7 +150,7 @@ def inference(
148
  with gr.Blocks(css=css) as app:
149
  gr.Markdown(
150
  '''
151
- <center><h1>Hussein Diffusion</h1></span>
152
  <span font-size:16px;">Generate stunning high quality illusion artwork with Stable Diffusion</span>
153
  </center>
154
  '''
@@ -158,7 +160,7 @@ with gr.Blocks(css=css) as app:
158
  with gr.Column():
159
  control_image = gr.Image(label="Input Illusion", type="pil", elem_id="control_image")
160
  controlnet_conditioning_scale = gr.Slider(minimum=0.0, maximum=5.0, step=0.01, value=0.8, label="Illusion strength", elem_id="illusion_strength", info="ControlNet conditioning scale")
161
- gr.Examples(examples=["checkers.png", "checkers_mid.jpg", "pattern.png", "ultra_checkers.png", "spiral.jpeg", "funky.jpeg" ], inputs=control_image)
162
  prompt = gr.Textbox(label="Prompt", elem_id="prompt")
163
  negative_prompt = gr.Textbox(label="Negative Prompt", value="low quality", elem_id="negative_prompt")
164
  with gr.Accordion(label="Advanced Options", open=False):
 
18
  from gallery_history import fetch_gallery_history, show_gallery_history
19
  from illusion_style import css
20
 
21
+ #BASE_MODEL = "HusseinHE/icbinh"
22
+
23
+ BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
24
 
25
  # Initialize both pipelines
26
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
 
150
  with gr.Blocks(css=css) as app:
151
  gr.Markdown(
152
  '''
153
+ <center><h1>Poiesis Diffusion</h1></span>
154
  <span font-size:16px;">Generate stunning high quality illusion artwork with Stable Diffusion</span>
155
  </center>
156
  '''
 
160
  with gr.Column():
161
  control_image = gr.Image(label="Input Illusion", type="pil", elem_id="control_image")
162
  controlnet_conditioning_scale = gr.Slider(minimum=0.0, maximum=5.0, step=0.01, value=0.8, label="Illusion strength", elem_id="illusion_strength", info="ControlNet conditioning scale")
163
+ gr.Examples(examples=["checkers.png", "checkers_mid.jpg", "pattern.png", "ultra_checkers.png", "spiral.jpeg", "funky.jpeg" , "poiesis.png" ], inputs=control_image)
164
  prompt = gr.Textbox(label="Prompt", elem_id="prompt")
165
  negative_prompt = gr.Textbox(label="Negative Prompt", value="low quality", elem_id="negative_prompt")
166
  with gr.Accordion(label="Advanced Options", open=False):