Spaces:

huggan
/

BigGAN

Runtime error

Jezia commited on Mar 26, 2022

Commit

bce2c8c

1 Parent(s): ad645e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from pytorch_pretrained_biggan import (BigGAN, one_hot_from_names, truncated_noi
                                        save_as_images, display_in_terminal)
 initial_archi = 'biggan-deep-128' #@param ['biggan-deep-128', 'biggan-deep-256', 'biggan-deep-512'] {allow-input: true}
-gan_model = BigGAN.from_pretrained(initial_archi).cuda().eval()
 # Prepare a input
 truncation = 0.4

                                        save_as_images, display_in_terminal)
 initial_archi = 'biggan-deep-128' #@param ['biggan-deep-128', 'biggan-deep-256', 'biggan-deep-512'] {allow-input: true}
+gan_model = BigGAN.from_pretrained(initial_archi)
 # Prepare a input
 truncation = 0.4