Update app.py

app.py

@@ -43,18 +43,18 @@ def on_model_change(model):
         if torch.cuda.is_available():
             pipe = pipe.to("cuda")
 
-def inference(prompt, guidance, steps):
+def inference(prompt, guidance, steps, width_input, height_input):
 
     promptPrev = prompt
     prompt = prompt_prefixes[current_model] + prompt
-    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512)
+    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
     image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
     return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(placeholder=promptPrev)
 
-def inference_example(prompt, guidance, steps):
+def inference_example(prompt, guidance, steps, width_input, height_input):
 
     prompt = prompt_prefixes[current_model] + prompt
-    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512)
+    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=width_input, height=height_input)
     image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
     return image
 
@@ -131,6 +131,8 @@ with gr.Blocks(css=css) as demo:
             prompt = gr.Textbox(label="Prompt", placeholder="{} is added automatically".format(prompt_prefixes[current_model]), elem_id="input-prompt")
             guidance = gr.Slider(label="Guidance scale", value=7, maximum=8)
             steps = gr.Slider(label="Steps", value=20, maximum=30, minimum=2)
+            width_input = gr.Slider(label="Width", value=512, maximum=960, minimum=360)
+            height_input = gr.Slider(label="Height", value=512, maximum=960, minimum=360)
             run = gr.Button(value="Run")
             gr.Markdown(f"Running on: {device}")
         with gr.Column():
@@ -143,16 +145,16 @@ with gr.Blocks(css=css) as demo:
           share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)
           
     model.change(on_model_change, inputs=model, outputs=[])
-    run.click(inference, inputs=[prompt, guidance, steps], outputs=[image_out, share_button, community_icon, loading_icon, prompt])
+    run.click(inference, inputs=[prompt, guidance, steps, width_input, height_input], outputs=[image_out, share_button, community_icon, loading_icon, prompt])
     
     share_button.click(None, [], [], _js=share_js)
     
     gr.Examples([
-        ["portrait of a beautiful fancy gorgeous anime girl, intricate details", 7, 20],
-        ["a beautiful perfect face girl, Anime fine details portrait of school girl in front of modern tokyo city landscape on the background deep bokeh, anime masterpiece by studio ghibli, 8k, sharp high quality anime, artstation", 7, 20],
-        ["city landscape with fancy car, racing on the road, gopro, intricate details, 4k, cyberpunk", 7, 20],
-        ["portrait of liu yifei girl, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20],
-        ["portrait of a muscular beard male in dgs illustration style, half-body, holding robot arms, strong chest", 7, 20],
+        ["portrait of a beautiful fancy gorgeous anime girl, intricate details", 7, 20, 704, 704],
+        ["a beautiful perfect face girl, Anime fine details portrait of school girl in front of modern tokyo city landscape on the background deep bokeh, anime masterpiece by studio ghibli, 8k, sharp high quality anime, artstation", 7, 20, 704, 704],
+        ["city landscape with fancy car, racing on the road, gopro, intricate details, 4k, cyberpunk", 7, 20, 704, 704],
+        ["portrait of liu yifei girl, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20, 704, 704],
+        ["portrait of a muscular beard male in dgs illustration style, half-body, holding robot arms, strong chest", 7, 20, 704, 704],
     ], [prompt, guidance, steps], image_out, inference_example, cache_examples=torch.cuda.is_available())
     gr.Markdown('''
       Models and Space by [@DGSpitzer](https://www.youtube.com/channel/UCzzsYBF4qwtMwJaPJZ5SuPg)❤️ [@大谷的游戏创作小屋](https://space.bilibili.com/176003)