bhadresh-savani commited on
Commit
bc6702c
1 Parent(s): a5a5107

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ transform = Compose(
12
  Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
13
  ]
14
  )
15
- model = GeneratorUNet.from_pretrained('huggan/pix2pix-cityscapes')
16
 
17
  def predict_fn(img):
18
  inp = transform(img).unsqueeze(0)
 
12
  Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
13
  ]
14
  )
15
+ model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')
16
 
17
  def predict_fn(img):
18
  inp = transform(img).unsqueeze(0)