Ryukijano commited on
Commit
8d56d1d
1 Parent(s): 8c5c8af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ def load_model(model_name):
10
  model_name = "Ryukijano/controlnet-fill-circle"
11
  tokenizer, controlnet, pipeline = load_model(model_name)
12
 
13
- def infer_fill_circle(prompt, negative_prompt, image):
14
  # Your inference function for fill circle control
15
  inputs = tokenizer(prompt, return_tensors="jax")
16
  # Implement your image preprocessing here
@@ -27,7 +27,7 @@ with gr.Blocks(theme='gradio/soft') as demo:
27
  fill_circle_input = gr.Image(label="Input Image")
28
  fill_circle_output = gr.Image(label="Output Image")
29
  submit_btn = gr.Button(value="Submit")
30
- fill_circle_inputs = [prompt_input_fill_circle, negative_prompt_fill_circle, fill_circle_input]
31
  submit_btn.click(fn=infer_fill_circle, inputs=fill_circle_inputs, outputs=[fill_circle_output])
32
 
33
  demo.launch()
 
10
  model_name = "Ryukijano/controlnet-fill-circle"
11
  tokenizer, controlnet, pipeline = load_model(model_name)
12
 
13
+ def infer_fill_circle(prompt, image):
14
  # Your inference function for fill circle control
15
  inputs = tokenizer(prompt, return_tensors="jax")
16
  # Implement your image preprocessing here
 
27
  fill_circle_input = gr.Image(label="Input Image")
28
  fill_circle_output = gr.Image(label="Output Image")
29
  submit_btn = gr.Button(value="Submit")
30
+ fill_circle_inputs = [prompt_input_fill_circle, fill_circle_input]
31
  submit_btn.click(fn=infer_fill_circle, inputs=fill_circle_inputs, outputs=[fill_circle_output])
32
 
33
  demo.launch()