Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -37,10 +37,7 @@ def generate_image(uploaded_image):
|
|
37 |
image = Image.open(uploaded_image)
|
38 |
return output
|
39 |
|
40 |
-
def infer(init_img, prompt, negative_prompt, seed,
|
41 |
-
|
42 |
-
if randomize_seed:
|
43 |
-
seed = random.randint(0, MAX_SEED)
|
44 |
|
45 |
generator = torch.Generator().manual_seed(seed)
|
46 |
|
@@ -162,7 +159,7 @@ with gr.Blocks(css=css) as demo:
|
|
162 |
|
163 |
run_button.click(
|
164 |
fn = infer,
|
165 |
-
inputs = [init_img, prompt, negative_prompt, seed,
|
166 |
outputs = [result]
|
167 |
)
|
168 |
|
|
|
37 |
image = Image.open(uploaded_image)
|
38 |
return output
|
39 |
|
40 |
+
def infer(init_img, prompt, negative_prompt, seed, width, height, guidance_scale, strength):
|
|
|
|
|
|
|
41 |
|
42 |
generator = torch.Generator().manual_seed(seed)
|
43 |
|
|
|
159 |
|
160 |
run_button.click(
|
161 |
fn = infer,
|
162 |
+
inputs = [init_img, prompt, negative_prompt, seed, width, height, guidance_scale, strength],
|
163 |
outputs = [result]
|
164 |
)
|
165 |
|