bryandts commited on
Commit
7c5a10d
1 Parent(s): c76f6d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 generator import Generator
12
 
13
  noise_dim = 16
14
  embed_dim = 384
15
  embed_out_dim = 256
16
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
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