Spaces:

huggan
/

Sketch2Shoes

Runtime error

bhadresh-savani commited on Apr 12, 2022

Commit

bc6702c

•

1 Parent(s): a5a5107

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ transform = Compose(
         Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
     ]
 )
-model = GeneratorUNet.from_pretrained('huggan/pix2pix-cityscapes')
 def predict_fn(img):
     inp = transform(img).unsqueeze(0)

         Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
     ]
 )
+model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')
 def predict_fn(img):
     inp = transform(img).unsqueeze(0)