Update app.py

app.py

@@ -8,12 +8,12 @@ import matplotlib.pyplot as plt
 import random
 import torch
 import torch.nn as nn
-from generator import Generator
+from generatorModel import Generator
 
 noise_dim = 16
 embed_dim = 384
 embed_out_dim = 256
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = 'cpu'
 
 generator = Generator(channels=3, embed_dim=embed_dim, noise_dim=noise_dim, embed_out_dim=embed_out_dim).to(device)