Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
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,
|
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()
|