myn0908 commited on
Commit
1d69f9a
1 Parent(s): 5efc806

update code

Browse files
Files changed (2) hide show
  1. S2I/commons/controller.py +1 -1
  2. app.py +2 -2
S2I/commons/controller.py CHANGED
@@ -58,7 +58,7 @@ class Sketch2ImageController():
58
  self.load_pipeline(zero_options=options)
59
  # prompt = prompt_template.replace("{prompt}", prompt)
60
 
61
- if input_type:
62
  img = image["composite"]
63
  else:
64
  img = Image.fromarray(np.array(image["composite"])[:, :, -1])
 
58
  self.load_pipeline(zero_options=options)
59
  # prompt = prompt_template.replace("{prompt}", prompt)
60
 
61
+ if type_flag:
62
  img = image["composite"]
63
  else:
64
  img = Image.fromarray(np.array(image["composite"])[:, :, -1])
app.py CHANGED
@@ -220,9 +220,9 @@ with gr.Blocks(css=css, theme="NoCrypt/miku@1.2.1") as demo:
220
  layers=False
221
  )
222
  with gr.Group():
 
 
223
  with gr.Row():
224
- input_type = gr.Checkbox(label="Use URL Image Sketch", value=False)
225
- url_image = gr.Textbox(label="Image URLS", value="")
226
  type_image = gr.Radio(
227
  choices=["RGB", "SKETCH"],
228
  value="SKETCH",
 
220
  layers=False
221
  )
222
  with gr.Group():
223
+ input_type = gr.Checkbox(label="Use URL Image Sketch", value=False)
224
+ url_image = gr.Textbox(label="Image URLS", value="")
225
  with gr.Row():
 
 
226
  type_image = gr.Radio(
227
  choices=["RGB", "SKETCH"],
228
  value="SKETCH",