DJStomp commited on
Commit
7e8f122
·
verified ·
1 Parent(s): a8df89e

(Hopefully) Final Touches

Browse files

- Fix an issue with label erroneously provided to gr.State initialization
- Minor adjustments and optimizations

Files changed (1) hide show
  1. app.py +7 -12
app.py CHANGED
@@ -1,11 +1,6 @@
1
  ##!/usr/bin/env python3
2
  # -*- coding: utf-8 -*-
3
  import os
4
-
5
- # print("Installing correct gradio version...")
6
- # os.system("pip install 'gradio>=3.50.0,<3.51.0' 'accelerate>=0.26.0' 'fsspec[http]==2024.6.1' --force-reinstall -q")
7
- # print("Installing Finished!")
8
-
9
  import gradio as gr
10
  import cv2
11
  from PIL import Image
@@ -173,9 +168,9 @@ with block:
173
  with gr.Tabs(elem_classes=["feedback"]):
174
  with gr.TabItem("Input Image"):
175
  input_image = gr.Image(type="numpy", label="input",scale=2, height=640)
176
- original_image = gr.State(value=None,label="index")
177
  original_mask = gr.State(value=None)
178
- selected_points = gr.State([],label="select points")
179
  with gr.Row(elem_id="Seg"):
180
  radio = gr.Radio(['foreground', 'background'], label='Click to seg: ', value='foreground',scale=2)
181
  undo_button = gr.Button('Undo seg', elem_id="btnSEG",scale=1)
@@ -184,7 +179,7 @@ with block:
184
  label="Negative Prompt",
185
  max_lines=5,
186
  placeholder="Please input your negative prompt",
187
- value='ugly, low quality',lines=1
188
  )
189
  with gr.Group():
190
  with gr.Row():
@@ -198,7 +193,7 @@ with block:
198
  )
199
  with gr.Group():
200
  seed = gr.Slider(
201
- label="Seed: ", minimum=0, maximum=2147483647, step=1, value=551793204
202
  )
203
  randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
204
 
@@ -209,12 +204,12 @@ with block:
209
  minimum=1,
210
  maximum=12,
211
  step=0.1,
212
- value=12,
213
  )
214
  num_inference_steps = gr.Slider(
215
  label="Number of inference steps",
216
  minimum=1,
217
- maximum=50,
218
  step=1,
219
  value=50,
220
  )
@@ -310,7 +305,7 @@ with block:
310
  # draw points
311
  output_mask = None
312
  if len(sel_pix) != 0:
313
- if isinstance(orig_img, int): # if orig_img is int, the image if select from examples
314
  temp = cv2.imread(image_examples[orig_img][0])
315
  temp = cv2.cvtColor(temp, cv2.COLOR_BGR2RGB)
316
  else:
 
1
  ##!/usr/bin/env python3
2
  # -*- coding: utf-8 -*-
3
  import os
 
 
 
 
 
4
  import gradio as gr
5
  import cv2
6
  from PIL import Image
 
168
  with gr.Tabs(elem_classes=["feedback"]):
169
  with gr.TabItem("Input Image"):
170
  input_image = gr.Image(type="numpy", label="input",scale=2, height=640)
171
+ original_image = gr.State(value=None)
172
  original_mask = gr.State(value=None)
173
+ selected_points = gr.State([])
174
  with gr.Row(elem_id="Seg"):
175
  radio = gr.Radio(['foreground', 'background'], label='Click to seg: ', value='foreground',scale=2)
176
  undo_button = gr.Button('Undo seg', elem_id="btnSEG",scale=1)
 
179
  label="Negative Prompt",
180
  max_lines=5,
181
  placeholder="Please input your negative prompt",
182
+ value='ugly, low quality, amateur, doodle, sketch',lines=1
183
  )
184
  with gr.Group():
185
  with gr.Row():
 
193
  )
194
  with gr.Group():
195
  seed = gr.Slider(
196
+ label="Seed: ", minimum=0, maximum=2147483647, step=1, value=694201337
197
  )
198
  randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
199
 
 
204
  minimum=1,
205
  maximum=12,
206
  step=0.1,
207
+ value=3.5,
208
  )
209
  num_inference_steps = gr.Slider(
210
  label="Number of inference steps",
211
  minimum=1,
212
+ maximum=100,
213
  step=1,
214
  value=50,
215
  )
 
305
  # draw points
306
  output_mask = None
307
  if len(sel_pix) != 0:
308
+ if isinstance(orig_img, int): # if orig_img is int, select the image from examples
309
  temp = cv2.imread(image_examples[orig_img][0])
310
  temp = cv2.cvtColor(temp, cv2.COLOR_BGR2RGB)
311
  else: