Spaces:

huggan
/

sim2real

Runtime error

Chris1 commited on Apr 17, 2022

Commit

6356cd8

1 Parent(s): b1c8049

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ transform = Compose([
     ])
-model = GeneratorResNet.from_pretrained('Chris1/sim2real', input_shape=(n_channels, image_size, image_size),
                 num_residual_blocks=9)
 gr.Interface(lambda image: pred_pipeline(image, transform),

     ])
+model = GeneratorResNet.from_pretrained('huggan/sim2real_cyclegan', input_shape=(n_channels, image_size, image_size),
                 num_residual_blocks=9)
 gr.Interface(lambda image: pred_pipeline(image, transform),