Spaces:
Sleeping
Sleeping
Commit
·
3927aba
1
Parent(s):
bba50e1
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ from torchvision.utils import save_image
|
|
| 14 |
class Generator(nn.Module):
|
| 15 |
def __init__(self, nc=4, nz=100, ngf=64):
|
| 16 |
super(Generator, self).__init__()
|
| 17 |
-
self.
|
| 18 |
nn.ConvTranspose2d(nz, ngf * 4, 3, 1, 0, bias=False),
|
| 19 |
nn.BatchNorm2d(ngf * 4),
|
| 20 |
nn.ReLU(True),
|
|
@@ -29,7 +29,7 @@ class Generator(nn.Module):
|
|
| 29 |
)
|
| 30 |
|
| 31 |
def forward(self, input):
|
| 32 |
-
output = self.
|
| 33 |
return output
|
| 34 |
|
| 35 |
|
|
|
|
| 14 |
class Generator(nn.Module):
|
| 15 |
def __init__(self, nc=4, nz=100, ngf=64):
|
| 16 |
super(Generator, self).__init__()
|
| 17 |
+
self.model = nn.Sequential(
|
| 18 |
nn.ConvTranspose2d(nz, ngf * 4, 3, 1, 0, bias=False),
|
| 19 |
nn.BatchNorm2d(ngf * 4),
|
| 20 |
nn.ReLU(True),
|
|
|
|
| 29 |
)
|
| 30 |
|
| 31 |
def forward(self, input):
|
| 32 |
+
output = self.model(input)
|
| 33 |
return output
|
| 34 |
|
| 35 |
|