Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ class Generator(nn.Module):
|
|
| 33 |
|
| 34 |
model = Generator(ngpu=0)
|
| 35 |
|
| 36 |
-
model.load_state_dict(torch.load('car_gen.pth'))
|
| 37 |
|
| 38 |
|
| 39 |
def generate(button):
|
|
|
|
| 33 |
|
| 34 |
model = Generator(ngpu=0)
|
| 35 |
|
| 36 |
+
model.load_state_dict(torch.load('car_gen.pth',map_location='cpu'))
|
| 37 |
|
| 38 |
|
| 39 |
def generate(button):
|