Update app.py

app.py

@@ -37,30 +37,30 @@ def layout_grid(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True
 network_pkl='braingan-400.pkl'
 with open(network_pkl, 'rb') as f:
     G = pickle.load(f)['G_ema'] 
-
+    
+device = torch.device('cuda')
+G.eval()
+G.to(device)
 def predict(Seed,choices):
-  device = torch.device('cuda')
-  G.eval()
-  G.to(device)
+
   shuffle_seed=None
   w_frames=60*4
   kind='cubic' 
   num_keyframes=None
   wraps=2
   psi=1 
-  device=torch.device('cuda')
 
   
   if choices=='4x2':
     grid_w = 4
     grid_h = 2
     s1=Seed
-    seeds=(np.arange(s1-16,s1)).tolist()
+    seeds=(np.arange(s1-8,s1)).tolist()
   if choices=='2x1':
     grid_w = 2
     grid_h = 1
     s1=Seed
-    seeds=(np.arange(s1-4,s1)).tolist()
+    seeds=(np.arange(s1-2,s1)).tolist()
 
 
   mp4='ex.mp4'