Update app.py

app.py

@@ -57,6 +57,9 @@ embed_dim = 384
 embed_out_dim = 256
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
+generator = Generator(channels=3, embed_dim=embed_dim, noise_dim=noise_dim, embed_out_dim=embed_out_dim).to(device)
+
+
 # Path to your .pth file
 gen_weight = 'generator_20240421_3.pth'