Norod78 commited on
Commit
21dde31
1 Parent(s): edb8225

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -28,7 +28,7 @@ def error_str(error, title="Error"):
28
  {error}""" if error else ""
29
 
30
  @spaces.GPU
31
- def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt=""):
32
 
33
  if torch.cuda.is_available():
34
  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
@@ -108,7 +108,7 @@ with gr.Blocks(css=css) as demo:
108
  prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2,placeholder="[your prompt]")
109
  generate = gr.Button(value="Generate")
110
 
111
- image_out = gr.Image(height=512)
112
  error_output = gr.Markdown()
113
 
114
  with gr.Column(scale=45):
@@ -118,17 +118,17 @@ with gr.Blocks(css=css) as demo:
118
 
119
  with gr.Row():
120
  guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)
121
- steps = gr.Slider(label="Steps", value=20, minimum=4, maximum=50, step=1)
122
 
123
  with gr.Row():
124
- width = gr.Slider(label="Width", value=512, minimum=64, maximum=1024, step=64)
125
- height = gr.Slider(label="Height", value=512, minimum=64, maximum=1024, step=64)
126
 
127
  seed = gr.Slider(0, 2147483647, label='Seed (0 = random)', value=0, step=1)
128
 
129
  with gr.Tab("Image to image"):
130
  with gr.Group():
131
- image = gr.Image(label="Image", height=256, type="pil")
132
  strength = gr.Slider(label="Transformation strength", minimum=0, maximum=1, step=0.01, value=0.5)
133
 
134
  inputs = [prompt, guidance, steps, width, height, seed, image, strength, neg_prompt]
 
28
  {error}""" if error else ""
29
 
30
  @spaces.GPU
31
+ def inference(prompt, guidance, steps, width=768, height=768, seed=0, img=None, strength=0.5, neg_prompt=""):
32
 
33
  if torch.cuda.is_available():
34
  generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
 
108
  prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2,placeholder="[your prompt]")
109
  generate = gr.Button(value="Generate")
110
 
111
+ image_out = gr.Image(height=768)
112
  error_output = gr.Markdown()
113
 
114
  with gr.Column(scale=45):
 
118
 
119
  with gr.Row():
120
  guidance = gr.Slider(label="Guidance scale", value=7.5, maximum=15)
121
+ steps = gr.Slider(label="Steps", value=24, minimum=4, maximum=50, step=1)
122
 
123
  with gr.Row():
124
+ width = gr.Slider(label="Width", value=768, minimum=64, maximum=1024, step=64)
125
+ height = gr.Slider(label="Height", value=768, minimum=64, maximum=1024, step=64)
126
 
127
  seed = gr.Slider(0, 2147483647, label='Seed (0 = random)', value=0, step=1)
128
 
129
  with gr.Tab("Image to image"):
130
  with gr.Group():
131
+ image = gr.Image(label="Image", height=768, type="pil")
132
  strength = gr.Slider(label="Transformation strength", minimum=0, maximum=1, step=0.01, value=0.5)
133
 
134
  inputs = [prompt, guidance, steps, width, height, seed, image, strength, neg_prompt]