Keshabwi66 commited on
Commit
54dbe52
·
verified ·
1 Parent(s): e67041d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
44
  return output_mask
45
 
46
 
47
- base_path = 'yisol/IDM-VTON'
48
 
49
  unet = UNet2DConditionModel.from_pretrained(
50
  base_path,
 
44
  return output_mask
45
 
46
 
47
+ base_path = 'Keshabwi66/SmartLugaModel'
48
 
49
  unet = UNet2DConditionModel.from_pretrained(
50
  base_path,