Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,12 +8,12 @@ import matplotlib.pyplot as plt
|
|
8 |
import random
|
9 |
import torch
|
10 |
import torch.nn as nn
|
11 |
-
from
|
12 |
|
13 |
noise_dim = 16
|
14 |
embed_dim = 384
|
15 |
embed_out_dim = 256
|
16 |
-
device =
|
17 |
|
18 |
generator = Generator(channels=3, embed_dim=embed_dim, noise_dim=noise_dim, embed_out_dim=embed_out_dim).to(device)
|
19 |
|
|
|
8 |
import random
|
9 |
import torch
|
10 |
import torch.nn as nn
|
11 |
+
from generatorModel import Generator
|
12 |
|
13 |
noise_dim = 16
|
14 |
embed_dim = 384
|
15 |
embed_out_dim = 256
|
16 |
+
device = 'cpu'
|
17 |
|
18 |
generator = Generator(channels=3, embed_dim=embed_dim, noise_dim=noise_dim, embed_out_dim=embed_out_dim).to(device)
|
19 |
|