craftgamesnetwork commited on
Commit
b9cfc71
·
verified ·
1 Parent(s): 22b3f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -59,8 +59,8 @@ def generate(
59
  use_img2img: bool = False,
60
  use_inpainting: bool = False,
61
  url = '',
62
- img_url = 'https://replicate.delivery/pbxt/JF3foGR90vm9BXSEXNaYkaeVKHYbJPinmpbMFvRtlDpH4MMk/out-0-1.png',
63
- mask_url = 'https://replicate.delivery/pbxt/JFIZFfJsSnWgxbTmEYLqhHIGdZo9o2BX3p47wSdn55HWtMON/mask1.png',
64
  ):
65
  if torch.cuda.is_available():
66
 
@@ -116,8 +116,8 @@ def generate(
116
  if use_inpainting:
117
  image = pipe(
118
  prompt=prompt,
119
- image_init=img_url,
120
- mask_image=mask_url,
121
  strength=strength_img2img,
122
  negative_prompt=negative_prompt,
123
  prompt_2=prompt_2,
 
59
  use_img2img: bool = False,
60
  use_inpainting: bool = False,
61
  url = '',
62
+ img_url = '',
63
+ mask_url = '',
64
  ):
65
  if torch.cuda.is_available():
66
 
 
116
  if use_inpainting:
117
  image = pipe(
118
  prompt=prompt,
119
+ image=image_init,
120
+ mask_image=mask_image,
121
  strength=strength_img2img,
122
  negative_prompt=negative_prompt,
123
  prompt_2=prompt_2,