Update app.py

app.py

@@ -54,92 +54,179 @@ css="""
 
 with gr.Blocks(css=css) as demo:
 
- # with gr.Tab("Text-to-Image"):
+ with gr.Tab("Text-to-Image"):
  
- with gr.Column(elem_id="col-container"):
- gr.Markdown(f"""
- # Text-to-Image Gradio Template
- """)
- 
- with gr.Row():
+ with gr.Column(elem_id="col-container"):
+ gr.Markdown(f"""
+ # Text-to-Image Gradio Template
+ """)
  
- prompt = gr.Text(
- label="Prompt",
- show_label=False,
- max_lines=1,
- placeholder="Enter your prompt",
- container=False,
- )
+ with gr.Row():
+ 
+ prompt = gr.Text(
+ label="Prompt",
+ show_label=False,
+ max_lines=1,
+ placeholder="Enter your prompt",
+ container=False,
+ )
+ 
+ run_button = gr.Button("Run", scale=0)
  
- run_button = gr.Button("Run", scale=0)
- 
- result = gr.Image(label="Result", show_label=False)
-
- with gr.Accordion("Advanced Settings", open=False):
+ result = gr.Image(label="Result", show_label=False)
+ 
+ with gr.Accordion("Advanced Settings", open=False):
+ 
+ negative_prompt = gr.Text(
+ label="Negative prompt",
+ max_lines=1,
+ placeholder="Enter a negative prompt",
+ visible=False,
+ )
+ 
+ seed = gr.Slider(
+ label="Seed",
+ minimum=0,
+ maximum=MAX_SEED,
+ step=1,
+ value=0,
+ )
+ 
+ randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+ 
+ with gr.Row():
+ 
+ width = gr.Slider(
+ label="Width",
+ minimum=256,
+ maximum=MAX_IMAGE_SIZE,
+ step=32,
+ value=1024, #Replace with defaults that work for your model
+ )
+ 
+ height = gr.Slider(
+ label="Height",
+ minimum=256,
+ maximum=MAX_IMAGE_SIZE,
+ step=32,
+ value=1024, #Replace with defaults that work for your model
+ )
+ 
+ with gr.Row():
+ 
+ guidance_scale = gr.Slider(
+ label="Guidance scale",
+ minimum=0.0,
+ maximum=10.0,
+ step=0.1,
+ value=0.0, #Replace with defaults that work for your model
+ )
+ 
+ num_inference_steps = gr.Slider(
+ label="Number of inference steps",
+ minimum=1,
+ maximum=50,
+ step=1,
+ value=2, #Replace with defaults that work for your model
+ )
  
- negative_prompt = gr.Text(
- label="Negative prompt",
- max_lines=1,
- placeholder="Enter a negative prompt",
- visible=False,
+ gr.Examples(
+ examples = examples,
+ inputs = [prompt]
  )
- 
- seed = gr.Slider(
- label="Seed",
- minimum=0,
- maximum=MAX_SEED,
- step=1,
- value=0,
+ 
+ run_button.click(
+ fn=infer,
+ inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+ outputs = [result, seed]
  )
- 
- randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+
+ with gr.Tab("Image-to-Image"):
+
+ with gr.Column(elem_id="col-container"):
+ gr.Markdown(f"""
+ # Text-to-Image Gradio Template
+ """)
  
  with gr.Row():
  
- width = gr.Slider(
- label="Width",
- minimum=256,
- maximum=MAX_IMAGE_SIZE,
- step=32,
- value=1024, #Replace with defaults that work for your model
+ prompt = gr.Text(
+ label="Prompt",
+ show_label=False,
+ max_lines=1,
+ placeholder="Enter your prompt",
+ container=False,
  )
  
- height = gr.Slider(
- label="Height",
- minimum=256,
- maximum=MAX_IMAGE_SIZE,
- step=32,
- value=1024, #Replace with defaults that work for your model
- )
+ run_button = gr.Button("Run", scale=0)
  
- with gr.Row():
+ result = gr.Image(label="Result", show_label=False)
+ 
+ with gr.Accordion("Advanced Settings", open=False):
  
- guidance_scale = gr.Slider(
- label="Guidance scale",
- minimum=0.0,
- maximum=10.0,
- step=0.1,
- value=0.0, #Replace with defaults that work for your model
+ negative_prompt = gr.Text(
+ label="Negative prompt",
+ max_lines=1,
+ placeholder="Enter a negative prompt",
+ visible=False,
  )
  
- num_inference_steps = gr.Slider(
- label="Number of inference steps",
- minimum=1,
- maximum=50,
+ seed = gr.Slider(
+ label="Seed",
+ minimum=0,
+ maximum=MAX_SEED,
  step=1,
- value=2, #Replace with defaults that work for your model
+ value=0,
  )
- 
- gr.Examples(
- examples = examples,
- inputs = [prompt]
- )
-
- run_button.click(
- fn=infer,
- inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
- outputs = [result, seed]
- )
+ 
+ randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+ 
+ with gr.Row():
+ 
+ width = gr.Slider(
+ label="Width",
+ minimum=256,
+ maximum=MAX_IMAGE_SIZE,
+ step=32,
+ value=1024, #Replace with defaults that work for your model
+ )
+ 
+ height = gr.Slider(
+ label="Height",
+ minimum=256,
+ maximum=MAX_IMAGE_SIZE,
+ step=32,
+ value=1024, #Replace with defaults that work for your model
+ )
+ 
+ with gr.Row():
+ 
+ guidance_scale = gr.Slider(
+ label="Guidance scale",
+ minimum=0.0,
+ maximum=10.0,
+ step=0.1,
+ value=7.5, #Replace with defaults that work for your model
+ )
+ 
+ num_inference_steps = gr.Slider(
+ label="Number of inference steps",
+ minimum=1,
+ maximum=50,
+ step=1,
+ value=25, #Replace with defaults that work for your model
+ )
+ 
+ gr.Examples(
+ examples = examples,
+ inputs = [prompt]
+ )
+ 
+ run_button.click(
+ fn=infer,
+ inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+ outputs = [result, seed]
+ )
  # gr.on(
  # triggers=[run_button.click, prompt.submit],
  # fn = infer,